change argument logic