From 511bd664900a6818e883b0b6eb2e3e4243efefac Mon Sep 17 00:00:00 2001
From: Jinzhu <wosmvp@gmail.com>
Date: Fri, 3 Apr 2020 07:15:30 +0800
Subject: [PATCH] Fix print code lines

---
 utils/utils.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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++ {