diff --git a/src/go/go.mod b/src/go/go.mod
index 4620bd2..46b2a4d 100644
--- a/src/go/go.mod
+++ b/src/go/go.mod
@@ -1,3 +1,3 @@
-module github.com/facebookresearch/gni/go
+module github.com/facebookresearch/gni/src/go
 
 go 1.18
diff --git a/src/go/main.go b/src/go/main.go
index f624f04..6f51c88 100644
--- a/src/go/main.go
+++ b/src/go/main.go
@@ -8,7 +8,7 @@ package main
 
 import (
 	"fmt"
-	gni "github.com/facebookresearch/gni/go"
+	gni "github.com/facebookresearch/gni/src/go"
 )
 
 func main() {