diff --git a/binarycpython/tests/tests_population_extensions/test_sampling_variables.py b/binarycpython/tests/tests_population_extensions/test_sampling_variables.py
index 953f5d1a9619504d61e71781416bdcc9724a8bd8..4e392d7848f9bb18887878cbaf786140d828b596 100644
--- a/binarycpython/tests/tests_population_extensions/test_sampling_variables.py
+++ b/binarycpython/tests/tests_population_extensions/test_sampling_variables.py
@@ -68,9 +68,15 @@ class test_add_sampling_variable(unittest.TestCase):
         self.assertIn("lnm1", test_pop.grid_options["_sampling_variables"])
         self.assertEqual(len(test_pop.grid_options["_sampling_variables"]), 2)
 
+
+class test_add_grid_variable(unittest.TestCase):
+    """
+    Unittests for add_sampling_variable function
+    """
+
     def test_add_grid_variable(self):
         with Capturing() as _:
-            self._test_add_sampling_variable()
+            self._test_add_grid_variable()
 
     def _test_add_grid_variable(self):
         """