Corrected function return type