diff --git a/data/datasets/stokes_cylinder/test_data.h5 b/data/datasets/stokes_cylinder/test_data.h5
index 3e200da..b0f32a7 100644
Binary files a/data/datasets/stokes_cylinder/test_data.h5 and b/data/datasets/stokes_cylinder/test_data.h5 differ
diff --git a/data/datasets/stokes_cylinder/train_data.h5 b/data/datasets/stokes_cylinder/train_data.h5
index 09748d7..cb14fcf 100644
Binary files a/data/datasets/stokes_cylinder/train_data.h5 and b/data/datasets/stokes_cylinder/train_data.h5 differ
diff --git a/physics_driven_ml/dataset_processing/pde_dataset.py b/physics_driven_ml/dataset_processing/pde_dataset.py
index e9b383d..da33d08 100644
--- a/physics_driven_ml/dataset_processing/pde_dataset.py
+++ b/physics_driven_ml/dataset_processing/pde_dataset.py
@@ -95,7 +95,7 @@ def load_dataset(self, fname: str):
             # Load adjacency list
             edge_index = np.array(afile.h5pyfile["edge_index"])
             # Load mesh
-            mesh = afile.load_mesh("mesh")
+            mesh = afile.load_mesh()
             # Load data
             for i in range(n):
                 f = afile.load_function(mesh, "f", idx=i)