diff --git a/services/directory/tests/FilterLists.Directory.Infrastructure.Migrations.Tests/SeedQueryDbContextTest.cs b/services/directory/tests/FilterLists.Directory.Infrastructure.Migrations.Tests/SeedQueryDbContextTest.cs
index 5671f4b73..befc73940 100644
--- a/services/directory/tests/FilterLists.Directory.Infrastructure.Migrations.Tests/SeedQueryDbContextTest.cs
+++ b/services/directory/tests/FilterLists.Directory.Infrastructure.Migrations.Tests/SeedQueryDbContextTest.cs
@@ -1,4 +1,5 @@
 using System.Threading.Tasks;
+using FilterLists.Directory.Infrastructure.Migrations.Migrations;
 using FilterLists.Directory.Infrastructure.Persistence.Queries.Context;
 using Microsoft.EntityFrameworkCore;
 using Xunit;
@@ -14,7 +15,7 @@ public async Task Migrate_DoesNotThrowException()
             {
                 const string connString = "Server=directory-db;Database=filterlists;Uid=filterlists;Pwd=filterlists;";
                 var options = new DbContextOptionsBuilder<QueryDbContext>()
-                    .UseNpgsql(connString, m => m.MigrationsAssembly("FilterLists.Directory.Infrastructure.Migrations"))
+                    .UseNpgsql(connString, m => m.MigrationsAssembly(typeof(Initial).Assembly.GetName().Name))
                     .EnableSensitiveDataLogging()
                     .Options;
                 await using var context = new QueryDbContext(options);