diff --git a/examples/frameworks/adk_demo/src/nat_adk_demo/nat_time_mcp_tool.py b/examples/frameworks/adk_demo/src/nat_adk_demo/nat_time_mcp_tool.py
index 5090d02bd..0fe0bbd59 100644
--- a/examples/frameworks/adk_demo/src/nat_adk_demo/nat_time_mcp_tool.py
+++ b/examples/frameworks/adk_demo/src/nat_adk_demo/nat_time_mcp_tool.py
@@ -19,6 +19,7 @@
 from zoneinfo import ZoneInfo
 
 from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
 from nat.builder.function_info import FunctionInfo
 from nat.cli.register_workflow import register_function
 from nat.data_models.function import FunctionBaseConfig
@@ -30,7 +31,7 @@ class TimeMCPToolConfig(FunctionBaseConfig, name="get_city_time_tool"):
     """Configuration for the get_city_time tool."""
 
 
-@register_function(config_type=TimeMCPToolConfig)
+@register_function(config_type=TimeMCPToolConfig, framework_wrappers=[LLMFrameworkEnum.ADK])
 async def get_city_time(_config: TimeMCPToolConfig, _builder: Builder) -> AsyncIterator[FunctionInfo]:
     """
     Register a get_city_time(city: str) -> str tool for ADK.
diff --git a/examples/frameworks/adk_demo/src/nat_adk_demo/weather_update_tool.py b/examples/frameworks/adk_demo/src/nat_adk_demo/weather_update_tool.py
index 9686d412d..48d26eb76 100644
--- a/examples/frameworks/adk_demo/src/nat_adk_demo/weather_update_tool.py
+++ b/examples/frameworks/adk_demo/src/nat_adk_demo/weather_update_tool.py
@@ -17,6 +17,7 @@
 from collections.abc import AsyncIterator
 
 from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
 from nat.builder.function_info import FunctionInfo
 from nat.cli.register_workflow import register_function
 from nat.data_models.function import FunctionBaseConfig
@@ -26,7 +27,7 @@ class WeatherToolConfig(FunctionBaseConfig, name="weather_update"):
     pass
 
 
-@register_function(config_type=WeatherToolConfig)
+@register_function(config_type=WeatherToolConfig, framework_wrappers=[LLMFrameworkEnum.ADK])
 async def weather_update(_config: WeatherToolConfig, _builder: Builder) -> AsyncIterator[FunctionInfo]:
 
     async def _weather_update(city: str) -> str: