Remove unnecessary if-else