diff --git a/watcher.go b/watcher.go
index 1956731..899ae25 100644
--- a/watcher.go
+++ b/watcher.go
@@ -11,15 +11,6 @@ import (
 
 	"github.com/casbin/casbin/persist"
 	"gocloud.dev/pubsub"
-
-	// Import the pubsub driver packages we want to be able to open.
-	_ "gocloud.dev/pubsub/awssnssqs"
-	_ "gocloud.dev/pubsub/azuresb"
-	_ "gocloud.dev/pubsub/gcppubsub"
-	_ "gocloud.dev/pubsub/kafkapubsub"
-	_ "gocloud.dev/pubsub/mempubsub"
-	_ "gocloud.dev/pubsub/natspubsub"
-	_ "gocloud.dev/pubsub/rabbitpubsub"
 )
 
 // check interface compatibility
diff --git a/watcher_test.go b/watcher_test.go
index 8528b39..440f889 100644
--- a/watcher_test.go
+++ b/watcher_test.go
@@ -10,6 +10,10 @@ import (
 	"github.com/casbin/casbin"
 	gnatsd "github.com/nats-io/nats-server/v2/test"
 	"github.com/nats-io/nats.go"
+
+	// Enable inmemory and NATS drivers
+	_ "github.com/rusenask/casbin-go-cloud-watcher/drivers/mempubsub"
+	_ "github.com/rusenask/casbin-go-cloud-watcher/drivers/natspubsub"
 )
 
 func TestNATSWatcher(t *testing.T) {