OnDismissCallback Refactor