From 49f63e257461b898d98924130d6b6b79e04c3ae7 Mon Sep 17 00:00:00 2001
From: Brevven <brevingame@gmail.com>
Date: Sat, 2 Jul 2022 00:15:06 -0700
Subject: [PATCH] subgroup

---
 changelog.txt           | 5 +++++
 info.json               | 2 +-
 zirconium-enriched.lua  | 2 +-
 zirconium-recipe-se.lua | 5 +++--
 4 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/changelog.txt b/changelog.txt
index 319fb63..3c9eef8 100644
--- a/changelog.txt
+++ b/changelog.txt
@@ -1,4 +1,9 @@
 ---------------------------------------------------------------------------------------------------
+Version: 0.6.2
+Date: 2021-07-02
+  Changes:
+    - SE K2 subgroup cleanup
+---------------------------------------------------------------------------------------------------
 Version: 0.6.1
 Date: 2021-07-02
   Changes:
diff --git a/info.json b/info.json
index 61aa180..15649fb 100644
--- a/info.json
+++ b/info.json
@@ -1,6 +1,6 @@
 {
     "name": "bzzirconium",
-    "version": "0.6.1",
+    "version": "0.6.2",
     "factorio_version": "1.1",
     "title": "Zirconium",
     "author": "Brevven",
diff --git a/zirconium-enriched.lua b/zirconium-enriched.lua
index 262becc..841da4c 100644
--- a/zirconium-enriched.lua
+++ b/zirconium-enriched.lua
@@ -24,6 +24,7 @@ data:extend({
     icon = "__bzzirconium__/graphics/icons/enriched-zircon.png",
     icon_size = 128,
     category = "chemistry",
+    main_product = "enriched-zircon",
     energy_required = 3,
     enabled = false,
     always_show_made_in = true,
@@ -48,7 +49,6 @@ data:extend({
       tertiary = {r = 0.690, g = 0.768, b = 0.870, a = 0.000}, 
       quaternary = {r = 0.0, g = 0.980, b = 0.603, a = 0.900}
     },
-    subgroup = "raw-material",
     order = "e03[enriched-zircon]"
   },
   {
diff --git a/zirconium-recipe-se.lua b/zirconium-recipe-se.lua
index 1b85fa5..e259b46 100644
--- a/zirconium-recipe-se.lua
+++ b/zirconium-recipe-se.lua
@@ -30,8 +30,8 @@ if string.sub(mods["space-exploration"], 1, 3) == "0.6" then
     order = "a-h-z-z",
   }
   })
-  data.raw.item["zirconia"].subgroup = "zirconium"
-  data.raw.item["zirconium-plate"].subgroup = "zirconium"
+  util.set_item_subgroup("zirconia", "zirconium")
+  util.set_item_subgroup("zirconium-plate", "zirconium")
   data:extend({
   {
     type = "item",
@@ -116,6 +116,7 @@ if string.sub(mods["space-exploration"], 1, 3) == "0.6" then
   util.add_effect("se-vulcanite-smelting", {type = "unlock-recipe", recipe= "zirconium-ingot"})
   util.add_effect("se-vulcanite-smelting", {type = "unlock-recipe", recipe= "zirconium-ingot-to-plate"})
   if mods["Krastorio2"] then
+    util.set_item_subgroup("enriched-zirconium", "zirconium")
     se_delivery_cannon_recipes["enriched-zirconium"] = {name= "enriched-zirconium"}
   else
     util.add_product("molten-zirconium", {"titanium-ore", 6})