Corrected function signature