1 files changed, 34 insertions, 14 deletions
diff --git a/src/main/java/pers/gwyog/gtneioreplugin/plugin/NEIPluginConfig.java b/src/main/java/pers/gwyog/gtneioreplugin/plugin/NEIPluginConfig.java
index 738834c8f0..76cd9c4d18 100644
--- a/src/main/java/pers/gwyog/gtneioreplugin/plugin/NEIPluginConfig.java
+++ b/src/main/java/pers/gwyog/gtneioreplugin/plugin/NEIPluginConfig.java
@@ -4,7 +4,14 @@ import codechicken.nei.api.API;
 import codechicken.nei.api.IConfigureNEI;
 import cpw.mods.fml.common.Loader;
 import pers.gwyog.gtneioreplugin.GTNEIOrePlugin;
-import pers.gwyog.gtneioreplugin.util.GTOreLayerHelper;
+import pers.gwyog.gtneioreplugin.plugin.gregtech5.PluginGT5AsteroidStat;
+import pers.gwyog.gtneioreplugin.plugin.gregtech5.PluginGT5IEVeinStat;
+import pers.gwyog.gtneioreplugin.plugin.gregtech5.PluginGT5SmallOreStat;
+import pers.gwyog.gtneioreplugin.plugin.gregtech5.PluginGT5VeinStat;
+import pers.gwyog.gtneioreplugin.plugin.gregtech6.PluginGT6BedrockOreStat;
+import pers.gwyog.gtneioreplugin.plugin.gregtech6.PluginGT6SmallOreStat;
+import pers.gwyog.gtneioreplugin.plugin.gregtech6.PluginGT6VeinStat;
+import pers.gwyog.gtneioreplugin.util.GT5OreLayerHelper;
 
 public class NEIPluginConfig implements IConfigureNEI {
 
@@ -20,20 +27,33 @@ public class NEIPluginConfig implements IConfigureNEI {
 
     @Override
     public void loadConfig() {
-        PluginVeinStat pluginVeinStat = new PluginVeinStat();
-        PluginAsteroidStat pluginAsteriodStat = new PluginAsteroidStat();
-        PluginSmallOreStat pluginSmallOreStat = new PluginSmallOreStat();
-        API.registerRecipeHandler(pluginVeinStat);
-        API.registerUsageHandler(pluginVeinStat);
-        API.registerRecipeHandler(pluginAsteriodStat);
-        API.registerUsageHandler(pluginAsteriodStat);
-        API.registerRecipeHandler(pluginSmallOreStat);
-        API.registerUsageHandler(pluginSmallOreStat);
-        if (GTOreLayerHelper.immersiveEngineeringSupport) {
-            PluginIEVeinStat pluginIEVeinStat = new PluginIEVeinStat();
-            API.registerRecipeHandler(pluginIEVeinStat);
-            API.registerUsageHandler(pluginIEVeinStat);
+        if (GTNEIOrePlugin.GTVersion.equals("GT5")) {
+            PluginGT5VeinStat pluginVeinStat = new PluginGT5VeinStat();
+            PluginGT5AsteroidStat pluginAsteriodStat = new PluginGT5AsteroidStat();
+            PluginGT5SmallOreStat pluginSmallOreStat = new PluginGT5SmallOreStat();
+            API.registerRecipeHandler(pluginVeinStat);
+            API.registerUsageHandler(pluginVeinStat);
+            API.registerRecipeHandler(pluginAsteriodStat);
+            API.registerUsageHandler(pluginAsteriodStat);
+            API.registerRecipeHandler(pluginSmallOreStat);
+            API.registerUsageHandler(pluginSmallOreStat);
+            if (GT5OreLayerHelper.immersiveEngineeringSupport) {
+                PluginGT5IEVeinStat pluginIEVeinStat = new PluginGT5IEVeinStat();
+                API.registerRecipeHandler(pluginIEVeinStat);
+                API.registerUsageHandler(pluginIEVeinStat);
+            }
         }
+        else {
+            PluginGT6VeinStat pluginGT6VeinStat = new PluginGT6VeinStat();
+            PluginGT6SmallOreStat pluginGT6SmallOreStat = new PluginGT6SmallOreStat();
+            PluginGT6BedrockOreStat pluginGT6BedrockOreStat = new PluginGT6BedrockOreStat();
+            API.registerRecipeHandler(pluginGT6VeinStat);
+            API.registerUsageHandler(pluginGT6VeinStat);
+            API.registerRecipeHandler(pluginGT6SmallOreStat);
+            API.registerUsageHandler(pluginGT6SmallOreStat);
+            API.registerRecipeHandler(pluginGT6BedrockOreStat);
+            API.registerUsageHandler(pluginGT6BedrockOreStat);
+        } 
     }
 
 }