Fix elif statement