Add missing condition