diff --git a/code/code.go b/code/code.go
index <HASH>..<HASH> 100644
--- a/code/code.go
+++ b/code/code.go
@@ -2,11 +2,12 @@ package code
 
 import (
 	"fmt"
-	"gentests/models"
 	"go/ast"
 	"go/parser"
 	"go/token"
 	"log"
+
+	"github.com/cweill/go-gentests/models"
 )
 
 func Parse(path string) *models.Info {
diff --git a/gentests.go b/gentests.go
index <HASH>..<HASH> 100644
--- a/gentests.go
+++ b/gentests.go
@@ -2,12 +2,13 @@ package main
 
 import (
 	"fmt"
-	"gentests/code"
-	"gentests/render"
 	"os"
 	"os/exec"
 	"path/filepath"
 	"strings"
+
+	"github.com/cweill/go-gentests/code"
+	"github.com/cweill/go-gentests/render"
 )
 
 func generateTestCases(f *os.File, path string) {
diff --git a/render/render.go b/render/render.go
index <HASH>..<HASH> 100644
--- a/render/render.go
+++ b/render/render.go
@@ -1,9 +1,10 @@
 package render
 
 import (
-	"gentests/models"
 	"io"
 	"text/template"
+
+	"github.com/cweill/go-gentests/models"
 )
 
 var tmpls = template.Must(template.ParseGlob("render/templates/*.tmpl"))
Update imports to point to github.com dir.