fix elif issue