diff --git a/pgw/PGW_Tests.ttcn b/pgw/PGW_Tests.ttcn
index 7bb6910..4730040 100644
--- a/pgw/PGW_Tests.ttcn
+++ b/pgw/PGW_Tests.ttcn
@@ -72,7 +72,7 @@
 	var PDU_DIAMETER msg;
 
 	if (DIAMETER_PROC.checkstate("Connected")) {
-		f_diameter_expect(imsi);
+		f_diameter_expect_imsi(imsi);
 	}
 
 	while (true) {