diff --git a/tests/plugins/document/dynamodb/dynamodb_test.go b/tests/plugins/document/dynamodb/dynamodb_test.go
index 344e96e18..bd979ab34 100644
--- a/tests/plugins/document/dynamodb/dynamodb_test.go
+++ b/tests/plugins/document/dynamodb/dynamodb_test.go
@@ -20,7 +20,7 @@ import (
 	"os/exec"
 
 	ds_plugin "github.com/nitric-dev/membrane/plugins/document/dynamodb"
-	"github.com/nitric-dev/membrane/plugins/document/test"
+	test "github.com/nitric-dev/membrane/tests/plugins/document"
 	. "github.com/onsi/ginkgo"
 
 	"github.com/aws/aws-sdk-go/aws"
diff --git a/tests/plugins/document/firestore/firestore_test.go b/tests/plugins/document/firestore/firestore_test.go
index 22ad7ebc7..7e195067d 100644
--- a/tests/plugins/document/firestore/firestore_test.go
+++ b/tests/plugins/document/firestore/firestore_test.go
@@ -23,7 +23,7 @@ import (
 
 	"cloud.google.com/go/firestore"
 	ds_plugin "github.com/nitric-dev/membrane/plugins/document/firestore"
-	"github.com/nitric-dev/membrane/plugins/document/test"
+	test "github.com/nitric-dev/membrane/tests/plugins/document"
 	. "github.com/onsi/ginkgo"
 )