Fix incorrect parameter type