Remove if-else construction