Fix wrong parameter issue