diff --git a/SceneManager/EntryPoint.cs b/SceneManager/EntryPoint.cs
index 1fcc7a3..80dddfe 100644
--- a/SceneManager/EntryPoint.cs
+++ b/SceneManager/EntryPoint.cs
@@ -60,9 +60,9 @@ namespace SceneManager
             }
 
             // Clean up cones
-            foreach (Rage.Object cone in BarrierMenu.barriers.Where(c => c))
+            foreach (Barrier barrier in BarrierMenu.barriers.Where(b => b.GetBarrier()))
             {
-                cone.Delete();
+                barrier.GetBarrier().Delete();
             }
             if (BarrierMenu.shadowBarrier)
             {
diff --git a/SceneManager/SceneManager.csproj b/SceneManager/SceneManager.csproj
index 0c98606..7af20fd 100644
--- a/SceneManager/SceneManager.csproj
+++ b/SceneManager/SceneManager.csproj
@@ -51,6 +51,7 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="AITasking.cs" />
+    <Compile Include="Barrier.cs" />
     <Compile Include="DebugGraphics.cs" />
     <Compile Include="Hints.cs" />
     <Compile Include="Menus\BarrierMenu.cs" />