Corrected the retrun type