Fix, shouldn't have used else if