Correct return value to match function definition