diff --git a/EFPowerToolsExample/EFPowerToolsExample.csproj b/EFPowerToolsExample/EFPowerToolsExample.csproj
index 5cf4644..3604760 100644
--- a/EFPowerToolsExample/EFPowerToolsExample.csproj
+++ b/EFPowerToolsExample/EFPowerToolsExample.csproj
@@ -1,5 +1,6 @@
 <?xml version="1.0" encoding="utf-8"?>
 <Project ToolsVersion="14.0" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
+  <Import Project="../packages/EntityFramework.6.5.1/build/EntityFramework.props" Condition="Exists('../packages/EntityFramework.6.5.1/build/EntityFramework.props')" />
   <Import Project="$(MSBuildExtensionsPath)\$(MSBuildToolsVersion)\Microsoft.Common.props" Condition="Exists('$(MSBuildExtensionsPath)\$(MSBuildToolsVersion)\Microsoft.Common.props')" />
   <PropertyGroup>
     <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
@@ -34,6 +35,14 @@
     <WarningLevel>4</WarningLevel>
   </PropertyGroup>
   <ItemGroup>
+    <Reference Include="EntityFramework, Version=6.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089">
+      <HintPath>..\packages\EntityFramework.6.5.1\lib\net45\EntityFramework.dll</HintPath>
+      <Private>True</Private>
+    </Reference>
+    <Reference Include="EntityFramework.SqlServer, Version=6.0.0.0, Culture=neutral, PublicKeyToken=b77a5c561934e089">
+      <HintPath>..\packages\EntityFramework.6.5.1\lib\net45\EntityFramework.SqlServer.dll</HintPath>
+      <Private>True</Private>
+    </Reference>
     <Reference Include="System" />
     <Reference Include="System.Core" />
     <Reference Include="System.Xml.Linq" />
@@ -51,6 +60,14 @@
     <None Include="App.config" />
   </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
+  <Target Name="EnsureNuGetPackageBuildImports" BeforeTargets="PrepareForBuild">
+    <PropertyGroup>
+      <ErrorText>This project references NuGet package(s) that are missing on this computer. Enable NuGet Package Restore to download them.  For more information, see http://go.microsoft.com/fwlink/?LinkID=322105.The missing file is {0}.</ErrorText>
+    </PropertyGroup>
+    <Error Condition="!Exists('../packages/EntityFramework.6.5.1/build/EntityFramework.props')" Text="$([System.String]::Format('$(ErrorText)', '../packages/EntityFramework.6.5.1/build/EntityFramework.props'))" />
+    <Error Condition="!Exists('../packages/EntityFramework.6.5.1/build/EntityFramework.targets')" Text="$([System.String]::Format('$(ErrorText)', '../packages/EntityFramework.6.5.1/build/EntityFramework.targets'))" />
+  </Target>
+  <Import Project="../packages/EntityFramework.6.5.1/build/EntityFramework.targets" Condition="Exists('../packages/EntityFramework.6.5.1/build/EntityFramework.targets')" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.
   <Target Name="BeforeBuild">
diff --git a/EFPowerToolsExample/packages.config b/EFPowerToolsExample/packages.config
index 3424f94..34a1e55 100644
--- a/EFPowerToolsExample/packages.config
+++ b/EFPowerToolsExample/packages.config
@@ -1,4 +1,4 @@
 <?xml version="1.0" encoding="utf-8"?>
 <packages>
-  <package id="EntityFramework" version="6.4.4" targetFramework="net452" />
+  <package id="EntityFramework" version="6.5.1" targetFramework="net452" />
 </packages>
\ No newline at end of file