Correct return type