diff --git a/plugin/deploy/bash_test.go b/plugin/deploy/bash_test.go
index 8af86661..ba502701 100644
--- a/plugin/deploy/bash_test.go
+++ b/plugin/deploy/bash_test.go
@@ -44,7 +44,7 @@ func setUpWithBash(input string) (string, error) {
 		return "", err
 	}
 	bf := buildfile.New()
-	buildStruct.Deploy.Write(bf)
+	buildStruct.Deploy.Write(bf, nil)
 	return bf.String(), err
 }
 
diff --git a/plugin/deploy/cloudfoundry_test.go b/plugin/deploy/cloudfoundry_test.go
index 91891f86..f8ba3988 100644
--- a/plugin/deploy/cloudfoundry_test.go
+++ b/plugin/deploy/cloudfoundry_test.go
@@ -57,7 +57,7 @@ func setUpWithCF(input string) (string, error) {
 		return "", err
 	}
 	bf := buildfile.New()
-	buildStruct.Deploy.Write(bf)
+	buildStruct.Deploy.Write(bf, nil)
 	return bf.String(), err
 }
 
diff --git a/plugin/deploy/ssh_test.go b/plugin/deploy/ssh_test.go
index 46ee3956..581f15cc 100644
--- a/plugin/deploy/ssh_test.go
+++ b/plugin/deploy/ssh_test.go
@@ -56,7 +56,7 @@ func setUp(input string) (string, error) {
 		return "", err
 	}
 	bf := buildfile.New()
-	buildStruct.Deploy.Write(bf)
+	buildStruct.Deploy.Write(bf, nil)
 	return bf.String(), err
 }