diff --git a/main.go b/main.go
index 6172e7ec..21af629c 100644
--- a/main.go
+++ b/main.go
@@ -523,7 +523,7 @@ func (s *DB) Rollback() *DB {
 }
 
 // WrapInTx wraps a method in a transaction
-func (s *DB) WrapInTx(f func(tx *gorm.DB) (error)) error {
+func (s *DB) WrapInTx(f func(tx *DB) (error)) error {
 	tx := s.Begin()
 	if err := f(tx); err != nil {
 		tx.Rollback()