diff --git a/examples/foundational/14s-function-calling-direct.py b/examples/foundational/14s-function-calling-direct.py
index 8c91a369e..7778461f6 100644
--- a/examples/foundational/14s-function-calling-direct.py
+++ b/examples/foundational/14s-function-calling-direct.py
@@ -10,7 +10,6 @@ import os
 from dotenv import load_dotenv
 from loguru import logger
 
-from pipecat.adapters.schemas.function_schema import FunctionSchema
 from pipecat.adapters.schemas.tools_schema import ToolsSchema
 from pipecat.audio.vad.silero import SileroVADAnalyzer
 from pipecat.frames.frames import TTSSpeakFrame
diff --git a/src/pipecat/adapters/schemas/direct_function.py b/src/pipecat/adapters/schemas/direct_function.py
index 377a8ea3e..54763c3d8 100644
--- a/src/pipecat/adapters/schemas/direct_function.py
+++ b/src/pipecat/adapters/schemas/direct_function.py
@@ -217,6 +217,7 @@ class DirectFunctionWrapper(BaseDirectFunctionWrapper):
     Wrapper around a DirectFunction that:
     - extracts metadata from the function signature and docstring
     - generates a corresponding FunctionSchema
+    - helps with function invocation
     """
 
     @classmethod