diff --git a/underthesea/datasets/uit_absa_hotel/uit_absa_hotel.py b/underthesea/datasets/uit_absa_hotel/uit_absa_hotel.py
index 93dc0219e..904a457ce 100644
--- a/underthesea/datasets/uit_absa_hotel/uit_absa_hotel.py
+++ b/underthesea/datasets/uit_absa_hotel/uit_absa_hotel.py
@@ -22,14 +22,13 @@ def encode(self, labels):
         if type(labels) == list:
             return [self.encode(label) for label in labels]
         label = labels  # label is a string
-        if label not in self.label2index:
-            index = self.vocab_size
-            self.label2index[label] = index
-            self.index2label[index] = label
-            self.vocab_size += 1
-            return index
-        else:
+        if label in self.label2index:
             return self.label2index[label]
+        index = self.vocab_size
+        self.label2index[label] = index
+        self.index2label[index] = label
+        self.vocab_size += 1
+        return index
 
 
 class UITABSAHotel:
@@ -41,7 +40,7 @@ def __init__(self, training="aspect"):
         train_file = join(DATASETS_FOLDER, UITABSAHotel.NAME, "Train_Hotel.txt")
         dev_file = join(DATASETS_FOLDER, UITABSAHotel.NAME, "Dev_Hotel.txt")
         test_file = join(DATASETS_FOLDER, UITABSAHotel.NAME, "Test_Hotel.txt")
-        print("Currently training: %s (aspect or polarity)" % training)
+        print(f"Currently training: {training} (aspect or polarity)")
 
         self.training = training  # aspect or polarity
         self.label_encoder = LabelEncoder()
diff --git a/underthesea/datasets/uit_absa_restaurant/uit_absa_restaurant.py b/underthesea/datasets/uit_absa_restaurant/uit_absa_restaurant.py
index 1ec08fed0..abd331f77 100644
--- a/underthesea/datasets/uit_absa_restaurant/uit_absa_restaurant.py
+++ b/underthesea/datasets/uit_absa_restaurant/uit_absa_restaurant.py
@@ -22,14 +22,13 @@ def encode(self, labels):
         if type(labels) == list:
             return [self.encode(label) for label in labels]
         label = labels  # label is a string
-        if label not in self.label2index:
-            index = self.vocab_size
-            self.label2index[label] = index
-            self.index2label[index] = label
-            self.vocab_size += 1
-            return index
-        else:
+        if label in self.label2index:
             return self.label2index[label]
+        index = self.vocab_size
+        self.label2index[label] = index
+        self.index2label[index] = label
+        self.vocab_size += 1
+        return index
 
 
 class UITABSARestaurant:
@@ -41,7 +40,7 @@ def __init__(self, training="aspect"):
         train_file = join(DATASETS_FOLDER, UITABSARestaurant.NAME, "Train.txt")
         dev_file = join(DATASETS_FOLDER, UITABSARestaurant.NAME, "Dev.txt")
         test_file = join(DATASETS_FOLDER, UITABSARestaurant.NAME, "Test.txt")
-        print("Currently training: %s (aspect or polarity)" % training)
+        print(f"Currently training: {training} (aspect or polarity)")
 
         self.training = training  # aspect or polarity
         self.label_encoder = LabelEncoder()