fix arg type error