Fixed incorrect function argument type