diff --git a/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/ModelExtensions.cs b/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/ModelExtensions.cs
index a44e9734f4..551397c23c 100644
--- a/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/ModelExtensions.cs
+++ b/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/ModelExtensions.cs
@@ -1,5 +1,8 @@
 // Copyright (c) Microsoft. All rights reserved.
 
+using Aspire.Hosting;
+using Aspire.Hosting.ApplicationModel;
+
 namespace AgentWebChat.AppHost;
 
 public static class ModelExtensions
diff --git a/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/Program.cs b/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/Program.cs
index 9b25fe3488..0e31a033ca 100644
--- a/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/Program.cs
+++ b/dotnet/samples/AgentWebChat/AgentWebChat.AppHost/Program.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft. All rights reserved.
 
 using AgentWebChat.AppHost;
+using Aspire.Hosting;
 
 var builder = DistributedApplication.CreateBuilder(args);