fix arg checking