Fix parameter issue