diff --git a/test/hook/context/context_combiner.go b/test/hook/context/context_combiner.go
index 9627c5aa..99a12c1d 100644
--- a/test/hook/context/context_combiner.go
+++ b/test/hook/context/context_combiner.go
@@ -7,7 +7,7 @@ import (
 	"github.com/deckhouse/module-sdk/pkg/hook/binding-context"
 	"github.com/deckhouse/module-sdk/pkg/hook/types"
 
-	"github.com/flant/shell-operator/pkg/hook/controller"
+	"github.com/deckhouse/module-sdk/pkg/hook/controller"
 	"github.com/flant/shell-operator/pkg/hook/task_metadata"
 	shell_operator "github.com/flant/shell-operator/pkg/shell-operator"
 	"github.com/flant/shell-operator/pkg/task"
diff --git a/test/hook/context/generator.go b/test/hook/context/generator.go
index 5b2587ee..653faff1 100644
--- a/test/hook/context/generator.go
+++ b/test/hook/context/generator.go
@@ -11,9 +11,9 @@ import (
 	"github.com/deckhouse/module-sdk/pkg/hook/types"
 	kubeeventsmanager "github.com/deckhouse/module-sdk/pkg/kube-events-manager"
 
+	"github.com/deckhouse/module-sdk/pkg/hook"
+	"github.com/deckhouse/module-sdk/pkg/hook/controller"
 	"github.com/flant/kube-client/fake"
-	"github.com/flant/shell-operator/pkg/hook"
-	"github.com/flant/shell-operator/pkg/hook/controller"
 	schedulemanager "github.com/flant/shell-operator/pkg/schedule_manager"
 )