Correct place to return

@tormozz48