Add fixes to get_return_url