Fix parameter handling