replace version check with arg check