if, elif, else fix