Correct return condition