fix arg type