diff --git a/session_delete.go b/session_delete.go index 8c9193c2..1c458fe1 100644 --- a/session_delete.go +++ b/session_delete.go @@ -226,7 +226,7 @@ func (session *Session) Delete(bean interface{}) (int64, error) { session.afterDeleteBeans[bean] = &afterClosures } } else { - if _, ok := interface{}(bean).(AfterInsertProcessor); ok { + if _, ok := interface{}(bean).(AfterDeleteProcessor); ok { session.afterDeleteBeans[bean] = nil } } diff --git a/session_update.go b/session_update.go index 17e5672e..27e2deb0 100644 --- a/session_update.go +++ b/session_update.go @@ -333,7 +333,7 @@ func (session *Session) Update(bean interface{}, condiBean ...interface{}) (int6 } } else { - if _, ok := interface{}(bean).(AfterInsertProcessor); ok { + if _, ok := interface{}(bean).(AfterUpdateProcessor); ok { session.afterUpdateBeans[bean] = nil } }