fixing a missing condition