Add missing protected blocks
Small fix