diff --git a/test/TestMLPotential.py b/test/TestMLPotential.py
index 7d6027e..9a0f1eb 100644
--- a/test/TestMLPotential.py
+++ b/test/TestMLPotential.py
@@ -9,7 +9,7 @@
 platform_ints = range(mm.Platform.getNumPlatforms())
 
 
-@pytest.mark.parametrize("implementation,platform_name", list(itertools.product(['nnpops', 'torchani'], list(platform_ints))))
+@pytest.mark.parametrize("implementation,platform_int", list(itertools.product(['nnpops', 'torchani'], list(platform_ints))))
 class TestMLPotential:
 
     def testCreateMixedSystem(self, implementation, platform_int):