From dc4a2cb096b3fd7ddb7d133853f6a5436026b2ca Mon Sep 17 00:00:00 2001
From: Soopyboo32 <49228220+Soopyboo32@users.noreply.github.com>
Date: Fri, 28 Jan 2022 15:18:15 +0800
Subject: fix bug

---
 featureClass/class.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'featureClass')

diff --git a/featureClass/class.js b/featureClass/class.js
index 581f6ba..d1ddbc6 100644
--- a/featureClass/class.js
+++ b/featureClass/class.js
@@ -76,17 +76,18 @@ class Feature {
         let theEvent
         try{
         theEvent = this.FeatureManager.registerForge(event, func, this)
-
-        this.forgeEvents[theEvent.id] = theEvent
         }catch(e){
             if(!messageIfError) messageIfError = "An error occured while registering the event " + event.class.toString().split(".").pop() + ", this may cause " + this.constructor.name + " to not work properly."
             ChatLib.chat(this.FeatureManager.messagePrefix + messageIfError)
         }
+        if(theEvent)this.forgeEvents[theEvent.id] = theEvent
 
         return theEvent
     }
 
     unregisterForge(event){
+        if(!event) return
+        
         this.FeatureManager.unregisterForge(event)
 
         delete this.forgeEvents[event.id]
-- 
cgit