diff --git a/samples/Sample.Migrations/DefaultDesignTimeDbContextFactory.cs b/samples/Sample.Migrations/DefaultDesignTimeDbContextFactory.cs
index 8514236b..89af9e21 100644
--- a/samples/Sample.Migrations/DefaultDesignTimeDbContextFactory.cs
+++ b/samples/Sample.Migrations/DefaultDesignTimeDbContextFactory.cs
@@ -41,9 +41,9 @@ namespace Sample.Migrations
                    op.UseShardingMigrationConfigure(op =>
                    {
                        op.ReplaceService<IMigrationsSqlGenerator,
-                           ShardingSqlServerMigrationsSqlGenerator<DefaultShardingTableDbContext>>();
+                           ShardingSqlServerMigrationsSqlGenerator>();
                    });
-                }).ReplaceService<ITableEnsureManager,SqlServerTableEnsureManager>().EnsureConfig();
+                }).EnsureConfig();
             _serviceProvider = services.BuildServiceProvider();
         }
 
diff --git a/samples/Sample.Migrations/EFCores/ShardingSqlServerMigrationsSqlGenerator.cs b/samples/Sample.Migrations/EFCores/ShardingSqlServerMigrationsSqlGenerator.cs
index ea3751cd..f7341521 100644
--- a/samples/Sample.Migrations/EFCores/ShardingSqlServerMigrationsSqlGenerator.cs
+++ b/samples/Sample.Migrations/EFCores/ShardingSqlServerMigrationsSqlGenerator.cs
@@ -13,7 +13,7 @@ namespace Sample.Migrations.EFCores
     /// <summary>
     /// https://github.com/Coldairarrow/EFCore.Sharding/blob/master/src/EFCore.Sharding.SqlServer/ShardingSqlServerMigrationsSqlGenerator.cs
     /// </summary>
-    public class ShardingSqlServerMigrationsSqlGenerator<TShardingDbContext> : SqlServerMigrationsSqlGenerator where TShardingDbContext:DbContext,IShardingDbContext
+    public class ShardingSqlServerMigrationsSqlGenerator : SqlServerMigrationsSqlGenerator
     {
         private readonly IShardingRuntimeContext _shardingRuntimeContext;