diff --git a/python/agent-framework/sample-agent/agent.py b/python/agent-framework/sample-agent/agent.py
index 4cf64639..87715206 100644
--- a/python/agent-framework/sample-agent/agent.py
+++ b/python/agent-framework/sample-agent/agent.py
@@ -38,7 +38,6 @@
 # AgentFramework SDK
 from agent_framework import ChatAgent
 from agent_framework.azure import AzureOpenAIChatClient
-from agent_framework.observability import setup_observability
 
 # Agent Interface
 from agent_interface import AgentInterface
diff --git a/python/agent-framework/sample-agent/host_agent_server.py b/python/agent-framework/sample-agent/host_agent_server.py
index eef43751..5efe11f8 100644
--- a/python/agent-framework/sample-agent/host_agent_server.py
+++ b/python/agent-framework/sample-agent/host_agent_server.py
@@ -12,8 +12,7 @@
 from aiohttp.web_middlewares import middleware as web_middleware
 from dotenv import load_dotenv
 from agent_interface import AgentInterface, check_agent_inheritance
-from microsoft_agents.activity import load_configuration_from_env, Activity
-from microsoft_agents.activity_types import ActivityTypes
+from microsoft_agents.activity import load_configuration_from_env, Activity, ActivityTypes
 from microsoft_agents.authentication.msal import MsalConnectionManager
 from microsoft_agents.hosting.aiohttp import (
     CloudAdapter,
@@ -35,8 +34,9 @@
     NotificationTypes,
     AgentNotificationActivity,
     ChannelId,
-    EmailResponse,
 )
+from microsoft_agents_a365.notifications import EmailResponse
+
 from microsoft_agents_a365.observability.core.config import configure
 from microsoft_agents_a365.observability.core.middleware.baggage_builder import (
     BaggageBuilder,