diff --git a/lib/mage/go.mod b/lib/mage/go.mod
index c0523c8..75a01ea 100644
--- a/lib/mage/go.mod
+++ b/lib/mage/go.mod
@@ -1,4 +1,4 @@
-module github.com/Azure-Samples/aio-dev-toolbox/toolbox
+module github.com/Azure-Samples/aio-dev-toolbox/lib/mage
 
 go 1.21.3
 
diff --git a/samples/example/go.mod b/samples/example/go.mod
new file mode 100644
index 0000000..96332e1
--- /dev/null
+++ b/samples/example/go.mod
@@ -0,0 +1,3 @@
+module github.com/Azure-Samples/aio-dev-toolbox/samples/example
+
+go 1.21.3
\ No newline at end of file