Correct return type annotation