Correct the return type