Fixed an embarrassing if-else statement.