diff --git a/cloudstack/cloudstack.go b/cloudstack/cloudstack.go
index ddf022d..bae9bb3 100644
--- a/cloudstack/cloudstack.go
+++ b/cloudstack/cloudstack.go
@@ -39,7 +39,7 @@ import (
 	"strings"
 	"time"
 
-	gomock "github.com/golang/mock/gomock"
+        gomock "go.uber.org/mock/gomock"
 )
 
 // UnlimitedResourceID is a special ID to define an unlimited resource