diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_base.py b/libs/langchain/tests/unit_tests/schema/runnable/test_base.py
index dde620dc803..6f1d238a73e 100644
--- a/libs/langchain/tests/unit_tests/schema/runnable/test_base.py
+++ b/libs/langchain/tests/unit_tests/schema/runnable/test_base.py
@@ -16,7 +16,6 @@ EXPECTED_ALL = [
     "Input",
     "Output",
     "Other",
-    "AnyMessage",
 ]
 
 
diff --git a/libs/langchain/tests/unit_tests/schema/test_messages.py b/libs/langchain/tests/unit_tests/schema/test_messages.py
index 3c13488cace..a7370652ffd 100644
--- a/libs/langchain/tests/unit_tests/schema/test_messages.py
+++ b/libs/langchain/tests/unit_tests/schema/test_messages.py
@@ -3,6 +3,7 @@ from langchain.schema.messages import __all__
 EXPECTED_ALL = [
     "AIMessage",
     "AIMessageChunk",
+    "AnyMessage",
     "BaseMessage",
     "BaseMessageChunk",
     "ChatMessage",