diff --git a/src/ic-websocket-cdk/service.example.did b/src/ic-websocket-cdk/service.example.did
index 478789c..456a033 100644
--- a/src/ic-websocket-cdk/service.example.did
+++ b/src/ic-websocket-cdk/service.example.did
@@ -1,7 +1,7 @@
 import "./ws_types.did";
 
 // define your message type here
-type MyMessageType = {
+type MyMessageType = record {
   some_field : text;
 };
 
diff --git a/src/ic-websocket-cdk/ws_types.did b/src/ic-websocket-cdk/ws_types.did
index 6db065a..2fa715a 100644
--- a/src/ic-websocket-cdk/ws_types.did
+++ b/src/ic-websocket-cdk/ws_types.did
@@ -23,6 +23,7 @@ type CanisterOutputCertifiedMessages = record {
   messages : vec CanisterOutputMessage;
   cert : blob;
   tree : blob;
+  is_end_of_queue : bool;
 };
 
 type CanisterWsOpenArguments = record {