Fixed bug in arguments