diff --git a/tests/configuration_init_tests.py b/tests/configuration_init_tests.py
index 84e88df..b2f7e74 100644
--- a/tests/configuration_init_tests.py
+++ b/tests/configuration_init_tests.py
@@ -41,7 +41,7 @@
         pci = 3
         _set_pci(self.cfg, pci)
         self.assertEqual(
-            self.cfg.get_parameter(ParameterName.PCI), pci,
+            self.cfg.get_parameter(ParameterName.PCI1), pci,
             'PCI value should be same as what was set',
         )
         with self.assertRaises(ConfigurationError):
@@ -122,7 +122,7 @@
         # Check the ip and port are sort properly
         _set_s1_connection(self.cfg, mme_ip, mme_port)
         self.assertEqual(
-            self.cfg.get_parameter(ParameterName.MME_IP), mme_ip,
+            self.cfg.get_parameter(ParameterName.MME_ADDRESS), mme_ip,
             'Expected mme ip to be set',
         )
         self.assertEqual(