Remove conditional in after_success