diff --git a/SpaceWarp/API/Mod.cs b/SpaceWarp/API/Mod.cs
index a28c03c0..f174b6e3 100644
--- a/SpaceWarp/API/Mod.cs
+++ b/SpaceWarp/API/Mod.cs
@@ -1,10 +1,11 @@
-using SpaceWarp.API.Configuration;
+using KSP.Game;
+using SpaceWarp.API.Configuration;
 using SpaceWarp.API.Logging;
 using UnityEngine;
 
 namespace SpaceWarp.API
 {
-	public abstract class Mod : MonoBehaviour
+	public abstract class Mod : KerbalMonoBehaviour
 	{
 		public BaseModLogger Logger;
 		public SpaceWarpManager Manager;
diff --git a/SpaceWarp/API/SpaceWarpManager.cs b/SpaceWarp/API/SpaceWarpManager.cs
index 64297b54..7a1612e5 100644
--- a/SpaceWarp/API/SpaceWarpManager.cs
+++ b/SpaceWarp/API/SpaceWarpManager.cs
@@ -4,6 +4,7 @@
 using System.Linq;
 using System.Reflection;
 using HarmonyLib;
+using KSP.Game;
 using UnityEngine;
 using Newtonsoft.Json;
 using SpaceWarp.API.Configuration;
@@ -15,7 +16,7 @@ namespace SpaceWarp.API
     /// <summary>
     /// Handles all the SpaceWarp initialization and mod processing.
     /// </summary>
-    public class SpaceWarpManager : MonoBehaviour
+    public class SpaceWarpManager : KerbalMonoBehaviour
     {
         private BaseModLogger _modLogger;