Fixed the else if