diff --git a/IOT_LGen_Steps.ttcn b/IOT_LGen_Steps.ttcn
index 63b86db..305991e 100644
--- a/IOT_LGen_Steps.ttcn
+++ b/IOT_LGen_Steps.ttcn
@@ -526,7 +526,7 @@
               hostName := vl_host.charstringVal,
               portNumber := vl_localPort
             },
-            proto := { udp := {} }
+            proto := { ip := { udp := {} } }
           },
           v_COAP_EntityCtxDB.data[v_COAP_ctx.eCtxIdx].localAddressIdx
         );
@@ -1266,9 +1266,9 @@
         if (v_COAP_FsmCtxDB.data[v_COAP_ctx.fsmCtxIdx].remoteAddressIdx >= 0)
         {
           v_COAP_transportEndpointDB.data[v_COAP_EntityCtxDB.data[v_COAP_ctx.eCtxIdx].localAddressIdx].proto :=
-            { dtls := { udp := {} }};
+            { ip := { dtls := { udp := {} }}};
           v_COAP_transportEndpointDB.data[v_COAP_FsmCtxDB.data[v_COAP_ctx.fsmCtxIdx].remoteAddressIdx].proto :=
-            { dtls := { udp := {} }};
+            { ip := { dtls := { udp := {} }}};
         }
         // We also need to initialize the DTLS connection:
         //  - we need to set the identity and the keys first into the fsmCtx