Fix wrong argument being passed