fix working with arguments