diff --git a/src/griml/test/test.py b/src/griml/test/test.py
index db5c0a5..ccb13cf 100644
--- a/src/griml/test/test.py
+++ b/src/griml/test/test.py
@@ -44,6 +44,7 @@ def create_sample_pointfile(self, filepath, num_features=5):
         gdf.reset_index(drop=True, inplace=True)
         gdf.set_index("row_id", inplace=True)
         gdf.to_file(filepath)
+        # return gdf
 
     def create_sample_polyfile(self, filepath, num_features=5, side_length=1.0):
         # Generate a synthetic GeoDataFrame with square polygon geometries
@@ -72,6 +73,7 @@ def create_sample_polyfile(self, filepath, num_features=5, side_length=1.0):
         gdf.reset_index(drop=True, inplace=True)
         gdf.set_index("row_id", inplace=True)
         gdf.to_file(filepath)
+        # return gdf
     
     def test_convert(self):
         '''Test vector to raster conversion'''