Fixing a missing condition