diff --git a/docker_test.go b/docker_test.go
index cba894e..5b3bccb 100644
--- a/docker_test.go
+++ b/docker_test.go
@@ -161,7 +161,7 @@ func TestCommandBuild(t *testing.T) {
 		tc := tc
 
 		t.Run(tc.name, func(t *testing.T) {
-			cmd := commandBuild(tc.build)
+			cmd := commandBuild(tc.build, "plugins/drone-docker:latest")
 
 			if !reflect.DeepEqual(cmd.String(), tc.want.String()) {
 				t.Errorf("Got cmd %v, want %v", cmd, tc.want)