diff --git a/ActiveStruts/ActiveStruts.csproj b/ActiveStruts/ActiveStruts.csproj
index db78906..158dbe7 100644
--- a/ActiveStruts/ActiveStruts.csproj
+++ b/ActiveStruts/ActiveStruts.csproj
@@ -45,9 +45,6 @@
       <HintPath>..\..\KSP_LIB\UnityEngine.dll</HintPath>
       <Private>False</Private>
     </Reference>
-    <Reference Include="KSPUtil">
-      <HintPath>..\..\KSP_LIB\KSPUtil.dll</HintPath>
-    </Reference>
     <Reference Include="UnityEngine.UI">
       <HintPath>..\..\KSP_LIB\UnityEngine.UI.dll</HintPath>
     </Reference>
diff --git a/ActiveStruts/Util/Utilities.cs b/ActiveStruts/Util/Utilities.cs
index a53d3ca..7c85889 100644
--- a/ActiveStruts/Util/Utilities.cs
+++ b/ActiveStruts/Util/Utilities.cs
@@ -4,6 +4,7 @@
 using ActiveStruts.Modules;
 using UnityEngine;
 using Object = UnityEngine.Object;
+using KSP.UI.Screens;
 
 namespace ActiveStruts.Util
 {
@@ -500,10 +501,12 @@ public static void RestoreVesselStaging(this Vessel vessel, List<PartStageBackup
             }
             if (addExtraStage)
             {
-                var currMax = Staging.lastStage;
-                Staging.AddStageAt(++currMax);
+                StageManager stManager = KSP.UI.Screens.StageManager.Instance;
+
+                var currMax = StageManager.LastStage;
+                stManager.AddStageAt(++currMax);
             }
-            vessel.currentStage = Staging.lastStage;
+            vessel.currentStage = StageManager.LastStage;
             FlightGlobals.SetActiveVessel(currVessel);
         }