diff --git a/user_implemented_agent.py b/user_implemented_agent.py
index 65a35c6..9db9ac1 100644
--- a/user_implemented_agent.py
+++ b/user_implemented_agent.py
@@ -1,11 +1,11 @@
 from vocode.user_implemented_agent.restful_agent import RESTfulAgent
 from vocode.user_implemented_agent.websocket_agent import WebSocketAgent 
 
-class EchoAgent(RESTfulAgent):
+class EchoAgent(WebSocketAgent):
 
     async def respond(self, input: str) -> str:
         return input
     
 if __name__ == "__main__":
     agent = EchoAgent()
-    agent.run()
\ No newline at end of file
+    agent.run()
diff --git a/vocode/models/agent.py b/vocode/models/agent.py
index 0b9267a..2f573f5 100644
--- a/vocode/models/agent.py
+++ b/vocode/models/agent.py
@@ -44,8 +44,9 @@ class RESTfulUserImplementedAgentConfig(AgentConfig, type=AgentType.RESTFUL_USER
         method: str = "POST"
 
     respond: EndpointConfig
-    generate_response: Optional[EndpointConfig]
-    update_last_bot_message_on_cut_off: Optional[EndpointConfig]
+    generate_responses: bool = False
+    # generate_response: Optional[EndpointConfig]
+    # update_last_bot_message_on_cut_off: Optional[EndpointConfig]
 
 class RESTfulAgentInput(BaseModel):
     human_input: str
@@ -58,8 +59,9 @@ class WebSocketUserImplementedAgentConfig(AgentConfig, type=AgentType.WEBSOCKET_
         url: str
 
     respond: RouteConfig
-    generate_response: Optional[RouteConfig]
-    send_message_on_cut_off: bool = False
+    generate_responses: bool = False
+    # generate_response: Optional[RouteConfig]
+    # send_message_on_cut_off: bool = False
 
 class WebSocketAgentMessageType(str, Enum):
     AGENT_BASE = 'agent_base'