From 7b892929cb3d9536f600fcbf92f8f89ed7daf0a4 Mon Sep 17 00:00:00 2001
From: Mister_Nebula <41904486+misternebula@users.noreply.github.com>
Date: Mon, 7 Mar 2022 20:52:28 +0000
Subject: [PATCH] Update QSBProbe.cs

---
 QSB/Tools/ProbeTool/QSBProbe.cs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/QSB/Tools/ProbeTool/QSBProbe.cs b/QSB/Tools/ProbeTool/QSBProbe.cs
index 3c9f50dc..d1b54613 100644
--- a/QSB/Tools/ProbeTool/QSBProbe.cs
+++ b/QSB/Tools/ProbeTool/QSBProbe.cs
@@ -4,7 +4,7 @@ using UnityEngine;
 
 namespace QSB.Tools.ProbeTool;
 
-public class QSBProbe : MonoBehaviour
+public class QSBProbe : MonoBehaviour, ILightSource
 {
 	public delegate void SurveyorProbeEvent();
 	public delegate void RetrieveEvent(float retrieveLength);
@@ -168,4 +168,7 @@ public class QSBProbe : MonoBehaviour
 
 		return false;
 	}
+
+	public LightSourceType GetLightSourceType() => LightSourceType.PROBE;
+	public OWLight2[] GetLights() => _illuminationCheckLights;
 }
\ No newline at end of file