diff --git a/src/samples/Sample23/ElsaContextFactory.cs b/src/samples/Sample23/ElsaContextFactory.cs
index d49e8b1e4..520651f97 100644
--- a/src/samples/Sample23/ElsaContextFactory.cs
+++ b/src/samples/Sample23/ElsaContextFactory.cs
@@ -1,4 +1,5 @@
 using Elsa;
+using Elsa.Persistence.EntityFrameworkCore.CustomSchema;
 using Elsa.Persistence.EntityFrameworkCore.DbContexts;
 using Elsa.Persistence.EntityFrameworkCore.Extensions;
 using Microsoft.EntityFrameworkCore;
diff --git a/src/samples/Sample23/Program.cs b/src/samples/Sample23/Program.cs
index 05ef67381..583b03b08 100644
--- a/src/samples/Sample23/Program.cs
+++ b/src/samples/Sample23/Program.cs
@@ -5,6 +5,7 @@ using Elsa.Extensions;
 using Elsa.Models;
 using Elsa.Persistence;
 using Elsa.Persistence.EntityFrameworkCore;
+using Elsa.Persistence.EntityFrameworkCore.CustomSchema;
 using Elsa.Persistence.EntityFrameworkCore.DbContexts;
 using Elsa.Persistence.EntityFrameworkCore.Extensions;
 using Elsa.Runtime;