diff --git a/pkg/controller/openpitrix/helmcategory/helm_category_controller_suite_test.go b/pkg/controller/openpitrix/helmcategory/helm_category_controller_suite_test.go
index b83ef5c6b..60bcd3c1c 100644
--- a/pkg/controller/openpitrix/helmcategory/helm_category_controller_suite_test.go
+++ b/pkg/controller/openpitrix/helmcategory/helm_category_controller_suite_test.go
@@ -81,7 +81,7 @@ var _ = BeforeSuite(func(done Done) {
 	})
 	Expect(err).ToNot(HaveOccurred())
 
-	err = Add(k8sManager)
+	err = (&ReconcileHelmCategory{}).SetupWithManager(k8sManager)
 	Expect(err).ToNot(HaveOccurred())
 
 	err = (&helmapplication.ReconcileHelmApplication{}).SetupWithManager(k8sManager)
diff --git a/pkg/controller/openpitrix/helmrepo/helm_repo_controller.go b/pkg/controller/openpitrix/helmrepo/helm_repo_controller.go
index 2d5383de1..cb2828195 100644
--- a/pkg/controller/openpitrix/helmrepo/helm_repo_controller.go
+++ b/pkg/controller/openpitrix/helmrepo/helm_repo_controller.go
@@ -86,6 +86,7 @@ type ReconcileHelmRepo struct {
 }
 
 func (r *ReconcileHelmRepo) SetupWithManager(mgr ctrl.Manager) error {
+	r.Client = mgr.GetClient()
 	r.scheme = mgr.GetScheme()
 	r.recorder = mgr.GetEventRecorderFor("workspace-controller")
 	r.config = mgr.GetConfig()
diff --git a/pkg/controller/openpitrix/helmrepo/helm_repo_controller_suite_test.go b/pkg/controller/openpitrix/helmrepo/helm_repo_controller_suite_test.go
index d14a54eb5..202b58bac 100644
--- a/pkg/controller/openpitrix/helmrepo/helm_repo_controller_suite_test.go
+++ b/pkg/controller/openpitrix/helmrepo/helm_repo_controller_suite_test.go
@@ -80,7 +80,7 @@ var _ = BeforeSuite(func(done Done) {
 	})
 	Expect(err).ToNot(HaveOccurred())
 
-	err = Add(k8sManager)
+	err = (&ReconcileHelmRepo{}).SetupWithManager(k8sManager)
 	Expect(err).ToNot(HaveOccurred())
 
 	go func() {