Fix if-else code style