Fixed return statement