diff --git a/samples/EShopSample/aspnet-core/src/EShopSample.DbMigrator/DbMigratorHostedService.cs b/samples/EShopSample/aspnet-core/src/EShopSample.DbMigrator/DbMigratorHostedService.cs
index 3fc10cbf..6fa0e8bf 100644
--- a/samples/EShopSample/aspnet-core/src/EShopSample.DbMigrator/DbMigratorHostedService.cs
+++ b/samples/EShopSample/aspnet-core/src/EShopSample.DbMigrator/DbMigratorHostedService.cs
@@ -6,6 +6,7 @@
 using Microsoft.Extensions.Hosting;
 using Serilog;
 using Volo.Abp;
+using Volo.Abp.Data;
 
 namespace EShopSample.DbMigrator;
 
@@ -23,11 +24,12 @@ public DbMigratorHostedService(IHostApplicationLifetime hostApplicationLifetime,
     public async Task StartAsync(CancellationToken cancellationToken)
     {
         using (var application = await AbpApplicationFactory.CreateAsync<EShopSampleDbMigratorModule>(options =>
-               {
-                   options.Services.ReplaceConfiguration(_configuration);
-                   options.UseAutofac();
-                   options.Services.AddLogging(c => c.AddSerilog());
-               }))
+        {
+            options.Services.ReplaceConfiguration(_configuration);
+            options.UseAutofac();
+            options.Services.AddLogging(c => c.AddSerilog());
+            options.AddDataMigrationEnvironment();
+        }))
         {
             await application.InitializeAsync();