Fixed a bug with arguments