Corrected a return type