Fix parameter type in the docs