Fix missing condition