fixing small bug in arguments