Fix type of argument