fix arguments in return statement