address pr comments

Signed-off-by: Raguideau <[email protected]>
HewlettPackard · Dec 8, 2023 · 1f775db · 1f775db
1 parent ba9c3e7
commit 1f775db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/aio5gc/context/session.go b/test/aio5gc/context/session.go
@@ -60,7 +60,7 @@ func (s *SessionContext) NewSessionContext() {
 	}}
 
 	s.lastAllocatedIP = net.ParseIP("10.0.0.1")
-	s.n3 = net.ParseIP("127.0.0.1")
+	s.n3 = net.ParseIP("127.0.0.1").To4()
 }
 
 func (s *SessionContext) GetN3() net.IP {
@@ -96,6 +96,6 @@ func (s *SessionContext) GetUnallocatedIP() net.IP {
 		log.Fatal("[5GC][NAS] Error while allocating ip for PDU session: " + err.Error())
 	}
 
-	s.lastAllocatedIP = net.ParseIP(ip)
+	s.lastAllocatedIP = net.ParseIP(ip).To4()
 	return s.lastAllocatedIP
 }