diff --git a/internal/test/mocks/mock_task_info.go b/internal/test/mocks/mock_task_info.go
index 5f8ba92..7529fa5 100644
--- a/internal/test/mocks/mock_task_info.go
+++ b/internal/test/mocks/mock_task_info.go
@@ -23,7 +23,7 @@
 	context "context"
 	reflect "reflect"
 
-	gomock "github.com/golang/mock/gomock"
+	gomock "go.uber.org/mock/gomock"
 )
 
 // MockTask is a mock of Task interface.