Fix in the return statement