diff --git a/pkg/controllers/state/suite_test.go b/pkg/controllers/state/suite_test.go
index b6207f0337..2025084a69 100644
--- a/pkg/controllers/state/suite_test.go
+++ b/pkg/controllers/state/suite_test.go
@@ -1483,7 +1483,7 @@ var _ = Describe("DaemonSet Controller", func() {
 			})
 		daemonsetPod1.Spec = daemonset.Spec.Template.Spec
 		ExpectApplied(ctx, env.Client, daemonsetPod1)
-		ExpectReconcileSucceeded(ctx, daemonsetController, client.ObjectKeyFromObject(daemonset))
+		ExpectReconcileSucceeded(ctx, kruiseDaemonSetController, client.ObjectKeyFromObject(daemonset))
 
 		Expect(cluster.GetDaemonSetPod(daemonset)).To(Equal(daemonsetPod1))