diff --git a/base/commands/migration/cancel_it_test.go b/base/commands/migration/cancel_it_test.go
index 980f3b47..25bb6bb1 100644
--- a/base/commands/migration/cancel_it_test.go
+++ b/base/commands/migration/cancel_it_test.go
@@ -60,6 +60,8 @@ func cancelTest(t *testing.T) {
 		})
 		setStatusCancelling(mID, tcx, ctx)
 		wg.Wait()
+		statusMap := MustValue(tcx.Client.GetMap(ctx, migration.StatusMapName))
+		MustValue(statusMap.Remove(ctx, mID))
 		tcx.AssertStdoutContains(`Migration canceled successfully.`)
 	})
 }