diff --git a/TrainerClass.py b/TrainerClass.py
index e8d0389..da2b153 100644
--- a/TrainerClass.py
+++ b/TrainerClass.py
@@ -540,7 +540,7 @@ class eNoseTrainer:
             if Y_discrete.ndim == 2:
                 Y_discrete = np.sum(Y_discrete, axis=1)
 
-            dataset = f'Conv1d-w{window}-{self.loader.smooth}'
+            dataset = f'Conv1D-w{window}-{self.loader.smooth}'
             for i, (train_index, test_index) in enumerate(gss.split(X_conv1d, Y_discrete, G_conv1d)):
                 self.logger.info("{:=^60}".format(f'CV {i+1}/{int(1/self.ratio)} {dataset}'))
 
@@ -554,7 +554,7 @@ class eNoseTrainer:
                 self.logger.debug(f"Y_test: {Y_test.shape}")
 
 
-                model_id = "Conv1d_v1_{}".format(i)
+                model_id = "Conv1D_v1_{}".format(i)
                 self.trained += 1
 
                 if self.row_exists(dataset, model_id):