diff --git a/build/build.cs b/build/build.cs
index 7c8f995a05..7694103e47 100644
--- a/build/build.cs
+++ b/build/build.cs
@@ -211,9 +211,7 @@ class Build : NukeBuild
         .ProceedAfterFailure()
         .Executes(() =>
         {
-            // TODO -- put codegen test back
-            //var codegenCommands = new[] { "codegen delete", "codegen write", "codegen test" };
-            var codegenCommands = new[] { "codegen delete", "codegen write" };
+            var codegenCommands = new[] { "codegen delete", "codegen write", "codegen test" };
             foreach (var command in codegenCommands)
             {
                 DotNetRun(s => s
diff --git a/src/Marten/Marten.csproj b/src/Marten/Marten.csproj
index 4b4147cd86..107f4db251 100644
--- a/src/Marten/Marten.csproj
+++ b/src/Marten/Marten.csproj
@@ -32,6 +32,7 @@
 
     <ItemGroup>
         <PackageReference Include="FSharp.Core" Version="9.0.100" />
+        <PackageReference Include="JasperFx" Version="0.7.1" />
         <PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
         <!-- This is forced by Npgsql peer dependency -->
         <PackageReference Include="Npgsql.Json.NET" Version="9.0.2" />