fixed wrong return statement