diff --git a/pina/data/base_dataset.py b/pina/data/base_dataset.py
index 2c28ba30..0f227ff0 100644
--- a/pina/data/base_dataset.py
+++ b/pina/data/base_dataset.py
@@ -116,7 +116,7 @@ def initialize(self):
                              dtype=torch.uint8)
                 for i in range(len(self.num_el_per_condition))
             ],
-                                               dim=0)
+                dim=0)
             for slot in self.__slots__:
                 current_attribute = getattr(self, slot)
                 if all(isinstance(a, LabelTensor) for a in current_attribute):
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
index 87fd9a15..c8e29cec 100644
--- a/tests/test_dataset.py
+++ b/tests/test_dataset.py
@@ -32,49 +32,49 @@ class Poisson(SpatialProblem):
 
     conditions = {
         'gamma1':
-        Condition(domain=CartesianDomain({
-            'x': [0, 1],
-            'y': 1
-        }),
-                  equation=FixedValue(0.0)),
+            Condition(domain=CartesianDomain({
+                'x': [0, 1],
+                'y': 1
+            }),
+                equation=FixedValue(0.0)),
         'gamma2':
-        Condition(domain=CartesianDomain({
-            'x': [0, 1],
-            'y': 0
-        }),
-                  equation=FixedValue(0.0)),
+            Condition(domain=CartesianDomain({
+                'x': [0, 1],
+                'y': 0
+            }),
+                equation=FixedValue(0.0)),
         'gamma3':
-        Condition(domain=CartesianDomain({
-            'x': 1,
-            'y': [0, 1]
-        }),
-                  equation=FixedValue(0.0)),
+            Condition(domain=CartesianDomain({
+                'x': 1,
+                'y': [0, 1]
+            }),
+                equation=FixedValue(0.0)),
         'gamma4':
-        Condition(domain=CartesianDomain({
-            'x': 0,
-            'y': [0, 1]
-        }),
-                  equation=FixedValue(0.0)),
+            Condition(domain=CartesianDomain({
+                'x': 0,
+                'y': [0, 1]
+            }),
+                equation=FixedValue(0.0)),
         'D':
-        Condition(input_points=LabelTensor(torch.rand(size=(100, 2)),
-                                           ['x', 'y']),
-                  equation=my_laplace),
+            Condition(input_points=LabelTensor(torch.rand(size=(100, 2)),
+                                               ['x', 'y']),
+                      equation=my_laplace),
         'data':
-        Condition(input_points=in_, output_points=out_),
+            Condition(input_points=in_, output_points=out_),
         'data2':
-        Condition(input_points=in2_, output_points=out2_),
+            Condition(input_points=in2_, output_points=out2_),
         'unsupervised':
-        Condition(
-            input_points=LabelTensor(torch.rand(size=(45, 2)), ['x', 'y']),
-            conditional_variables=LabelTensor(torch.ones(size=(45, 1)),
-                                              ['alpha']),
-        ),
+            Condition(
+                input_points=LabelTensor(torch.rand(size=(45, 2)), ['x', 'y']),
+                conditional_variables=LabelTensor(torch.ones(size=(45, 1)),
+                                                  ['alpha']),
+            ),
         'unsupervised2':
-        Condition(
-            input_points=LabelTensor(torch.rand(size=(90, 2)), ['x', 'y']),
-            conditional_variables=LabelTensor(torch.ones(size=(90, 1)),
-                                              ['alpha']),
-        )
+            Condition(
+                input_points=LabelTensor(torch.rand(size=(90, 2)), ['x', 'y']),
+                conditional_variables=LabelTensor(torch.ones(size=(90, 1)),
+                                                  ['alpha']),
+            )
     }
 
 
diff --git a/tests/test_solvers/test_supervised_solver.py b/tests/test_solvers/test_supervised_solver.py
index 8ceadcd9..ebe8179e 100644
--- a/tests/test_solvers/test_supervised_solver.py
+++ b/tests/test_solvers/test_supervised_solver.py
@@ -121,7 +121,7 @@ def test_train_cpu():
                       batch_size=5,
                       train_size=1,
                       test_size=0.,
-                      eval_size=0.)
+                      val_size=0.)
     trainer.train()
 test_train_cpu()