diff --git a/plugin_test.go b/plugin_test.go
index bc55c18..3006f08 100644
--- a/plugin_test.go
+++ b/plugin_test.go
@@ -17,7 +17,7 @@ func TestMissingConfig(t *testing.T) {
 func TestMissingSSHConfig(t *testing.T) {
 	plugin := Plugin{
 		Config: Config{
-			Host:     "example.com",
+			Host:     []string{"example.com"},
 			Username: "ubuntu",
 		},
 	}
@@ -30,7 +30,7 @@ func TestMissingSSHConfig(t *testing.T) {
 func TestMissingSourceConfig(t *testing.T) {
 	plugin := Plugin{
 		Config: Config{
-			Host:     "example.com",
+			Host:     []string{"example.com"},
 			Username: "ubuntu",
 			Port:     "443",
 			Password: "1234",