Fix more cases of the OnLoad issue

Author: JonnyOThan

Source/RocketSoundEnhancement/PartModules/RSE_Wheels.cs

@@ -33,6 +33,14 @@ public override void OnStart(StartState state)
             if (state == StartState.Editor || state == StartState.None)
                 return;
 
+            // This likely doesn't work correctly with B9PS if it's trying to modify the module config
+            if (part != null && part.partInfo != null && part.partInfo.partPrefab != null)
+            {
+                int moduleIndex = part.modules.IndexOf(this);
+                var prefab = part.partInfo.partPrefab.modules[moduleIndex] as RSE_Wheels;
+                offLoadVolumeScale = prefab.offLoadVolumeScale;
+            }
+
             base.OnStart(state);
 
             moduleWheel = part.GetComponent<ModuleWheelBase>();
@@ -176,11 +184,8 @@ public override void OnLoad(ConfigNode node)
         {
             base.OnLoad(node);
 
-            if (part?.partInfo?.partPrefab != null)
+            if (HighLogic.LoadedScene != GameScenes.LOADING)
             {
-                int moduleIndex = part.modules.IndexOf(this);
-                var prefab = part.partInfo.partPrefab.modules[moduleIndex] as RSE_Wheels;
-				offLoadVolumeScale = prefab.offLoadVolumeScale;
                 return;
             }
 

Source/RocketSoundEnhancement/PartModules/ShipEffectsCollisions.cs

@@ -32,6 +32,16 @@ public override void OnStart(StartState state)
             if (state == StartState.Editor || state == StartState.None)
                 return;
 
+            // This likely doesn't work correctly with B9PS if it's trying to modify the module config
+            if (part != null && part.partInfo != null && part.partInfo.partPrefab != null)
+            {
+                int moduleIndex = part.modules.IndexOf(this);
+                var prefab = part.partInfo.partPrefab.modules[moduleIndex] as ShipEffectsCollisions;
+
+                SoundLayerCollisionGroups = prefab.SoundLayerCollisionGroups;
+                return;
+            }
+
             base.OnStart(state);
 
             if (SoundLayerCollisionGroups.Count > 0)
@@ -120,12 +130,8 @@ public override void OnLoad(ConfigNode node)
         {
             base.OnLoad(node);
 
-            if (part?.partInfo?.partPrefab != null)
+            if (HighLogic.LoadedScene != GameScenes.LOADING)
             {
-                int moduleIndex = part.modules.IndexOf(this);
-				var prefab = part.partInfo.partPrefab.modules[moduleIndex] as ShipEffectsCollisions;
-
-				SoundLayerCollisionGroups = prefab.SoundLayerCollisionGroups;
                 return;
             }