diff --git a/internal/context/user_plane_information_test.go b/internal/context/user_plane_information_test.go
index 5e8fe47b..88fe24da 100644
--- a/internal/context/user_plane_information_test.go
+++ b/internal/context/user_plane_information_test.go
@@ -504,6 +504,7 @@ func TestGetUEIPPool(t *testing.T) {
 	userplaneInformation := smf_context.NewUserPlaneInformation(configForIPPoolAllocate)
 	for _, upf := range userplaneInformation.UPFs {
 		upf.UPFStatus = smf_context.AssociatedSetUpSuccess
+		upf.Association, upf.AssociationCancelFunc = context.WithCancel(context.Background())
 	}
 
 	for ci, tc := range testCasesOfGetUEIPPool {