Fix the return type