diff --git a/samples/ChatSample/Data/ApplicationDbContextFactory.cs b/samples/ChatSample/Data/ApplicationDbContextFactory.cs
index 37780f7ae0..4b91568b3a 100644
--- a/samples/ChatSample/Data/ApplicationDbContextFactory.cs
+++ b/samples/ChatSample/Data/ApplicationDbContextFactory.cs
@@ -3,12 +3,12 @@
 
 using System.IO;
 using Microsoft.EntityFrameworkCore;
-using Microsoft.EntityFrameworkCore.Infrastructure;
+using Microsoft.EntityFrameworkCore.Design;
 using Microsoft.Extensions.Configuration;
 
 namespace ChatSample.Data
 {
-    public class ApplicationDbContextFactory : IDbContextFactory<ApplicationDbContext>
+    public class ApplicationDbContextFactory : IDesignTimeDbContextFactory<ApplicationDbContext>
     {
         public ApplicationDbContext Create(string[] args)
         {