fix arg check