diff --git a/epyt/tests/test_unit_MSX.py b/epyt/tests/test_unit_MSX.py
index 7b31aed..3699b61 100644
--- a/epyt/tests/test_unit_MSX.py
+++ b/epyt/tests/test_unit_MSX.py
@@ -14,7 +14,7 @@ def setUp(self):
         inpname = os.path.join(DIRNAME, 'msx-examples', 'Net3-NH2CL.inp')
         msxfile = os.path.join(DIRNAME, 'msx-examples', 'Net3-NH2CL.msx')
         self.epanetClass = epanet(inpname, msx=True)
-        self.msxClass = epanetmsxapi(msxfile)
+        self.msxClass = self.epanetClass.loadMSXFile(msxfile)
 
     def tearDown(self):
         """Call after every test case."""
@@ -281,4 +281,4 @@ def test_MSXgetqual(self):
 
 
 if __name__ == "__main__":
-    unittest.main()  # run all tests
+    unittest.main()
\ No newline at end of file