diff --git a/README.md b/README.md
index 46e014b7c..1dcd82a5d 100644
--- a/README.md
+++ b/README.md
@@ -84,7 +84,7 @@ tool_node = ToolNode(tools)
 model = ChatAnthropic(model="claude-3-5-sonnet-20240620", temperature=0).bind_tools(tools)
 
 # Define the function that determines whether to continue or not
-def should_continue(state: MessagesState) -> Literal["tools", END]:
+def should_continue(state: MessagesState):
     messages = state['messages']
     last_message = messages[-1]
     # If the LLM makes a tool call, then we route to the "tools" node