diff --git a/api/server/sdk/volume_ops_test.go b/api/server/sdk/volume_ops_test.go
index c17a97228..efd9c4c84 100644
--- a/api/server/sdk/volume_ops_test.go
+++ b/api/server/sdk/volume_ops_test.go
@@ -151,7 +151,7 @@ func TestSdkVolumeCreateCheckIdempotencyWaitForReady(t *testing.T) {
 	s.MockDriver().
 		EXPECT().
 		Inspect(gomock.Any(), []string{name}).
-		Do(func([]string) {
+		Do(func(context.Context, []string) {
 			count++
 			if count == 4 {
 				vol.Status = api.VolumeStatus_VOLUME_STATUS_UP