diff --git a/Models/AccountState.cs b/src/PolyTrader.Core/Models/AccountState.cs
similarity index 98%
rename from Models/AccountState.cs
rename to src/PolyTrader.Core/Models/AccountState.cs
index 81f6c49..1a25364 100644
--- a/Models/AccountState.cs
+++ b/src/PolyTrader.Core/Models/AccountState.cs
@@ -1,6 +1,4 @@
 using System;
-using MongoDB.Driver;
-using PolyTraderSharp.Extensions;
 using System.ComponentModel;
 using System.Collections.Concurrent;
 using System.Linq;
diff --git a/Models/MarketData.cs b/src/PolyTrader.Core/Models/MarketData.cs
similarity index 100%
rename from Models/MarketData.cs
rename to src/PolyTrader.Core/Models/MarketData.cs
diff --git a/Models/Position.cs b/src/PolyTrader.Core/Models/Position.cs
similarity index 100%
rename from Models/Position.cs
rename to src/PolyTrader.Core/Models/Position.cs
diff --git a/src/PolyTrader.Core/PolyTrader.Core.csproj b/src/PolyTrader.Core/PolyTrader.Core.csproj
index 371d001..30d0206 100644
--- a/src/PolyTrader.Core/PolyTrader.Core.csproj
+++ b/src/PolyTrader.Core/PolyTrader.Core.csproj
@@ -6,4 +6,8 @@
     <Nullable>enable</Nullable>
   </PropertyGroup>
 
+  <ItemGroup>
+    <PackageReference Include="MongoDB.Driver" Version="2.24.0" />
+  </ItemGroup>
+
 </Project>