fixed bug with arguments