Fixed return type annotation