diff --git a/README.md b/README.md
index 4322e7e..b64f4a3 100644
--- a/README.md
+++ b/README.md
@@ -25,8 +25,9 @@ public class MyDbContextFactory : IDbContextFactory<MyDbContext>
         // LazyLoading specific
         if (_isLazy)
         {
-            dbContextOptionsBuilder.ReplaceService<Microsoft.EntityFrameworkCore.Metadata.Internal.IEntityMaterializerSource, Microsoft.EntityFrameworkCore.LazyLoading.Internal.LazyLoadingEntityMaterializerSource<MyDbContext>>();
+            dbContextOptionsBuilder.ReplaceService<Microsoft.EntityFrameworkCore.Metadata.Internal.IEntityMaterializerSource, Microsoft.EntityFrameworkCore.LazyLoading.Metadata.Internal.LazyLoadingEntityMaterializerSource<MyDbContext>>();
             dbContextOptionsBuilder.ReplaceService<Microsoft.EntityFrameworkCore.Internal.IConcurrencyDetector, Microsoft.EntityFrameworkCore.LazyLoading.Internal.ConcurrencyDetector>();
+            dbContextOptionsBuilder.ReplaceService<Microsoft.EntityFrameworkCore.Query.Internal.ICompiledQueryCache, Microsoft.EntityFrameworkCore.LazyLoading.Query.Internal.PerDbContextCompiledQueryCache>();
         }
             
 
@@ -36,7 +37,8 @@ public class MyDbContextFactory : IDbContextFactory<MyDbContext>
         // LazyLoading specific
         if (_isLazy)
         {
-            (ctx.GetService<Microsoft.EntityFrameworkCore.Metadata.Internal.IEntityMaterializerSource>() as Microsoft.EntityFrameworkCore.LazyLoading.Internal.LazyLoadingEntityMaterializerSource<MyDbContext>).SetDbContext(ctx);
+            (ctx.GetService<Microsoft.EntityFrameworkCore.Metadata.Internal.IEntityMaterializerSource>() as Microsoft.EntityFrameworkCore.LazyLoading.Metadata.Internal.LazyLoadingEntityMaterializerSource<MyDbContext>).SetDbContext(ctx);
+            (ctx.GetService<Microsoft.EntityFrameworkCore.Query.Internal.ICompiledQueryCache>() as Microsoft.EntityFrameworkCore.LazyLoading.Query.Internal.PerDbContextCompiledQueryCache).SetDbContext(ctx);
         }
 
         return ctx;
diff --git a/src/Microsoft.EntityFrameworkCore.LazyLoading/Microsoft.EntityFrameworkCore.LazyLoading.csproj b/src/Microsoft.EntityFrameworkCore.LazyLoading/Microsoft.EntityFrameworkCore.LazyLoading.csproj
index 96254f7..f5f5f36 100644
--- a/src/Microsoft.EntityFrameworkCore.LazyLoading/Microsoft.EntityFrameworkCore.LazyLoading.csproj
+++ b/src/Microsoft.EntityFrameworkCore.LazyLoading/Microsoft.EntityFrameworkCore.LazyLoading.csproj
@@ -2,13 +2,15 @@
 
   <PropertyGroup>
     <TargetFramework>netstandard1.3</TargetFramework>
-    <Version>1.1.1-beta1</Version>
+    <Version>1.1.1-beta2</Version>
     <Authors>darxis</Authors>
     <Company />
     <Description>This package enables LazyLoading for EntityFramework Core</Description>
     <PackageTags>Entity Framework Core EF EntityFramework LazyLoading Lazy Include DbContext EntityFrameworkCore</PackageTags>
     <RepositoryUrl>https://github.com/darxis/EntityFramework.LazyLoading</RepositoryUrl>
     <PackageProjectUrl>https://github.com/darxis/EntityFramework.LazyLoading</PackageProjectUrl>
+    <AssemblyVersion>1.1.1.1</AssemblyVersion>
+    <FileVersion>1.1.1.1</FileVersion>
   </PropertyGroup>
 
   <ItemGroup>