Fixed LocalJumpError: unexpected return