diff --git a/PHI-MB/main.lua b/PHI-MB/main.lua
index a474503..2fd1f99 100644
--- a/PHI-MB/main.lua
+++ b/PHI-MB/main.lua
@@ -15,10 +15,8 @@ local function EE(source, tier)
         local item = table.deepcopy(data.raw['assembling-machine']['assembling-machine-3'])
     elseif source == 'electric-furnace' then
         local item = table.deepcopy(data.raw['furnace']['electric-furnace'])
-    elseif source == 'chemical-plant' then
-        local item = table.deepcopy(data.raw['assembling-machine']['chemical-plant'])
-    elseif source == 'oil-refinery' then
-        local item = table.deepcopy(data.raw['assembling-machine']['oil-refinery'])
+    elseif (source == 'chemical-plant') or (source == 'oil-refinery') then
+        local item = table.deepcopy(data.raw['assembling-machine'][source])
     else
         local item = table.deepcopy(data.raw[source][source])
     end