diff --git a/dataManager/gFunDataset.py b/dataManager/gFunDataset.py
index 4242a3d..39c1539 100644
--- a/dataManager/gFunDataset.py
+++ b/dataManager/gFunDataset.py
@@ -57,7 +57,6 @@ class SimpleGfunDataset:
         print(f"tr: {tr} - va: {va} - te: {te}")
     
     def load_csv_inference(self):
-        # test = pd.read_csv(os.path.join(self.datadir, "test.small.csv" if not self.reduced else "test.small.csv"))
         test = pd.read_csv(self.datadir)
         self._set_labels(test)
         self._set_langs(train=None, test=test)
diff --git a/dataManager/utils.py b/dataManager/utils.py
index 9c86032..f6bf378 100644
--- a/dataManager/utils.py
+++ b/dataManager/utils.py
@@ -14,10 +14,10 @@ def load_from_pickle(path, dataset_name, nrows):
     return loaded
 
 
-def get_dataset(datasetp_path, args):
+def get_dataset(dataset_path, args):
     dataset = SimpleGfunDataset(
         dataset_name="rai",
-        datapath=datasetp_path,
+        datapath=dataset_path,
         textual=True,
         visual=False,
         multilabel=False,