Add missing protection