Fix wrong argument error