diff --git a/tests/hooks_test.go b/tests/hooks_test.go
index 0e6ab2fe..d2c387a3 100644
--- a/tests/hooks_test.go
+++ b/tests/hooks_test.go
@@ -254,6 +254,10 @@ func TestUseDBInHooks(t *testing.T) {
 		t.Fatalf("Failed to create product, got error: %v", err)
 	}
 
+	testPanic := Product2{Name: "testPanic", Price: 100}
+	DB.Create(testPanic)
+	DB.Model(Product2{}).Where("name", "testPanic").Update("name", "testPanic1")
+
 	var result Product2
 	if err := DB.First(&result, "name = ?", "Nice").Error; err != nil {
 		t.Fatalf("Failed to query product, got error: %v", err)
diff --git a/tests/update_test.go b/tests/update_test.go
index 819cf5e0..41ea5d27 100644
--- a/tests/update_test.go
+++ b/tests/update_test.go
@@ -763,7 +763,3 @@ func TestUpdateReturning(t *testing.T) {
 		t.Errorf("failed to return updated age column")
 	}
 }
-
-func TestAfterCreateWithPanic(t *testing.T)  {
-	DB.Model(User{}).Where("name=?", "update-1").Update("name", "update-1-1")
-}
diff --git a/utils/tests/models.go b/utils/tests/models.go
index 61e1165d..22e8e659 100644
--- a/utils/tests/models.go
+++ b/utils/tests/models.go
@@ -31,10 +31,6 @@ type User struct {
 	Active    bool
 }
 
-func (u *User) AfterCreate(tx *gorm.DB) error {
-	return nil
-}
-
 type Account struct {
 	gorm.Model
 	UserID sql.NullInt64