diff --git a/tests/envs/test_discrete.py b/tests/envs/test_discrete.py
index a1612636..e4c773a4 100644
--- a/tests/envs/test_discrete.py
+++ b/tests/envs/test_discrete.py
@@ -11,7 +11,7 @@ class TestDiscreteEnv(TestCase):
 
     def test_init_from_microgrid(self):
         microgrid = get_modular_microgrid()
-        env = DiscreteMicrogridEnv(microgrid)
+        env = DiscreteMicrogridEnv.from_microgrid(microgrid)
 
         self.assertEqual(env.modules, microgrid.modules)
         self.assertIsNot(env.modules.to_tuples(), microgrid.modules.to_tuples())