after_restart callback not called on rollback