diff --git a/src/MusicStore/Program.cs b/src/MusicStore/Program.cs
index 23812e98e1..7bc0f254a7 100644
--- a/src/MusicStore/Program.cs
+++ b/src/MusicStore/Program.cs
@@ -28,12 +28,13 @@ namespace MusicStore
 
             var serviceCollection = HostingServices.Create(_hostServiceProvider);
             var services = serviceCollection.BuildServiceProvider();
+            var applicationLifetime = services.GetRequiredService<IApplicationLifetime>();
 
             var context = new HostingContext()
             {
-                Services = services,
+                ApplicationLifetime = applicationLifetime,
                 Configuration = config,
-                ServerName = "Microsoft.AspNet.Server.WebListener",
+                ServerFactoryLocation = "Microsoft.AspNet.Server.WebListener",
                 ApplicationName = "MusicStore"
             };