diff --git a/utils/utils.go b/utils/utils.go
index 25cd585a..8521d09b 100644
--- a/utils/utils.go
+++ b/utils/utils.go
@@ -7,8 +7,8 @@ import (
 	"unicode"
 )
 
-var goSrcRegexp = regexp.MustCompile(`/gorm/.*.go`)
-var goTestRegexp = regexp.MustCompile(`/gorm/.*test.*.go`)
+var goSrcRegexp = regexp.MustCompile(`jinzhu/gorm(@.*)?/.*.go`)
+var goTestRegexp = regexp.MustCompile(`jinzhu/gorm(@.*)?/.*test.go`)
 
 func FileWithLineNum() string {
 	for i := 2; i < 15; i++ {