From 3c8eab8b9387a4d46a5e8ed6b7332f0ab0fe228c Mon Sep 17 00:00:00 2001
From: Noah Pilarski <mehmehmeh313@gmail.com>
Date: Sun, 19 Jun 2022 18:46:24 -0400
Subject: [PATCH] Set white hole in DB vessel

---
 NewHorizons/Handlers/VesselWarpHandler.cs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/NewHorizons/Handlers/VesselWarpHandler.cs b/NewHorizons/Handlers/VesselWarpHandler.cs
index 140b05f7..78352a8c 100644
--- a/NewHorizons/Handlers/VesselWarpHandler.cs
+++ b/NewHorizons/Handlers/VesselWarpHandler.cs
@@ -148,6 +148,13 @@ namespace NewHorizons.Handlers
             if (vesselWarpController._targetWarpPlatform != null)
                 vesselWarpController._targetWarpPlatform.OnReceiveWarpedBody += OnReceiveWarpedBody;
 
+            if (vesselWarpController._whiteHole == null)
+            {
+                GameObject whiteHole = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension/Sector_VesselBridge/Interactibles_VesselBridge/WhiteHole");
+                vesselWarpController._whiteHole = whiteHole.GetComponentInChildren<SingularityController>();
+                vesselWarpController._whiteHoleOneShot = vesselWarpController._whiteHole.transform.parent.Find("WhiteHoleAudio_OneShot").GetComponent<OWAudioSource>();
+            }
+
             Instance.ModHelper.Events.Unity.FireOnNextUpdate(() => SetupWarpController(vesselWarpController, true));
 
             return spawnPoint;