return fix

wrong return