Add missing flag check.