Add missing OS checks