Fix parameter type