diff --git a/mme/MME_Tests.ttcn b/mme/MME_Tests.ttcn
index 17a4d60..cc75377 100644
--- a/mme/MME_Tests.ttcn
+++ b/mme/MME_Tests.ttcn
@@ -361,7 +361,7 @@
 	g_Tguard.start(t_guard);
 	activate(as_Tguard());
 	if (DIAMETER_PROC.checkstate("Connected")) {
-		f_diameter_expect(g_pars.ue_pars.imsi);
+		f_diameter_expect_imsi(g_pars.ue_pars.imsi);
 	}
 	if (SGsAP_PROC.checkstate("Connected")) {
 		/* Route all SGsAP mesages for our IMSIto us */