diff --git a/executor_internal_test.go b/executor_internal_test.go
index 628b63155..0fa10f44e 100644
--- a/executor_internal_test.go
+++ b/executor_internal_test.go
@@ -585,12 +585,13 @@ func TestExecutor_DeleteRows(t *testing.T) {
 		t.Fatalf("failed to read row: %v", err)
 	}
 
-	changed, err := DeleteRows(row, idx, shard)
+	ctx := context.Background()
+	changed, err := DeleteRows(ctx, row, idx, shard)
 	if !changed || err != nil {
 		t.Fatalf("failed to delete row: %v", err)
 	}
 
-	changed, err = DeleteRows(row, idx, shard)
+	changed, err = DeleteRows(ctx, row, idx, shard)
 	if changed {
 		t.Fatalf("expected delete to not clear bit but it did")
 	}