Fixed issue in return, line 2