If-then-else case is now correct