diff --git a/api/queue/work/batch_deployment_test.go b/api/queue/work/batch_deployment_test.go
index 43baffd7b..433195db5 100644
--- a/api/queue/work/batch_deployment_test.go
+++ b/api/queue/work/batch_deployment_test.go
@@ -18,7 +18,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"k8s.io/apimachinery/pkg/api/resource"
-	"k8s.io/utils/clock"
+	clock "k8s.io/utils/clock/testing"
 )
 
 const (
diff --git a/api/service/prediction_job_service_test.go b/api/service/prediction_job_service_test.go
index 3235db198..3ffa50bf4 100644
--- a/api/service/prediction_job_service_test.go
+++ b/api/service/prediction_job_service_test.go
@@ -24,7 +24,7 @@ import (
 	"github.com/jinzhu/copier"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
-	"k8s.io/utils/clock"
+	clock "k8s.io/utils/clock/testing"
 
 	"github.com/caraml-dev/merlin/batch"
 	"github.com/caraml-dev/merlin/batch/mocks"