From baf44e326ac115090cfb57a7f76f6554a85de3ba Mon Sep 17 00:00:00 2001
From: yoyosource <yoyosource@nidido.de>
Date: Tue, 1 Feb 2022 09:00:01 +0100
Subject: [PATCH] Update some LinkageUtils problems

Signed-off-by: yoyosource <yoyosource@nidido.de>
---
 .../src/de/steamwar/bausystem/linkage/LinkageUtils.java   | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/BauSystem_Main/src/de/steamwar/bausystem/linkage/LinkageUtils.java b/BauSystem_Main/src/de/steamwar/bausystem/linkage/LinkageUtils.java
index 85f41d5a..825eff86 100644
--- a/BauSystem_Main/src/de/steamwar/bausystem/linkage/LinkageUtils.java
+++ b/BauSystem_Main/src/de/steamwar/bausystem/linkage/LinkageUtils.java
@@ -20,6 +20,8 @@
 package de.steamwar.bausystem.linkage;
 
 import de.steamwar.bausystem.BauSystem;
+import de.steamwar.bausystem.utils.FlatteningWrapper;
+import de.steamwar.bausystem.utils.NMSWrapper;
 import lombok.experimental.UtilityClass;
 import org.bukkit.Bukkit;
 
@@ -39,6 +41,12 @@ public class LinkageUtils {
     private Map<Class<?>, Object> objectMap = new HashMap<>();
     private Set<Field> fieldsToLink = new HashSet<>();
 
+    {
+        objectMap.put(BauSystem.class, BauSystem.getInstance());
+        objectMap.put(FlatteningWrapper.IFlatteningWrapper.class, FlatteningWrapper.impl);
+        objectMap.put(NMSWrapper.INMSWrapper.class, NMSWrapper.impl);
+    }
+
     public void link() {
         internalLinkOrUnlink(false, Linked.class);
         internalLinkFields();