fix arg problem