small return fix