forgot one if statement