Corrected return type