diff --git a/app/services/setting_test.go b/app/services/setting_test.go
index 7ffe0e2f..495fbe46 100644
--- a/app/services/setting_test.go
+++ b/app/services/setting_test.go
@@ -39,7 +39,7 @@ func (s *SettingTestSuite) TestGet() {
 
 func (s *SettingTestSuite) TestSet() {
 	mockOrm, mockDb, _, _ := mock.Orm()
-	mockOrm.On("Query").Return(mockDb).Twice()
+	mockOrm.On("Query").Return(mockDb)
 	mockDb.On("Where", "key", "test").Return(mockDb).Once()
 	mockDb.On("UpdateOrCreate", &models.Setting{}, models.Setting{Key: "test"}, models.Setting{Value: "test"}).Return(nil).Once()
 	err := s.setting.Set("test", "test")
diff --git a/app/services/user_test.go b/app/services/user_test.go
index 8acf85e7..e9bdb08d 100644
--- a/app/services/user_test.go
+++ b/app/services/user_test.go
@@ -27,7 +27,7 @@ func (s *UserTestSuite) SetupTest() {
 
 func (s *UserTestSuite) TestCreate() {
 	mockOrm, mockDb, _, _ := mock.Orm()
-	mockOrm.On("Query").Return(mockDb).Once()
+	mockOrm.On("Query").Return(mockDb)
 	mockDb.On("Create", &models.User{
 		Username: "haozi",
 		Password: "123456",