Fix minor bug with arguments