diff --git a/migrations/steel-display.lua b/migrations/steel-display.lua
index 46aa4fb..c61a4d3 100644
--- a/migrations/steel-display.lua
+++ b/migrations/steel-display.lua
@@ -1,11 +1,11 @@
+local DID = require("globals")
+
 for index, force in pairs(game.forces) do
-  local technologies = force.technologies
-  local recipes = force.recipes
-
-  if technologies["steel-processing"].researched then
-    recipes["steel-display-small"].enabled = true
-    recipes["steel-display-medium"].enabled = true
-    recipes["steel-display"].enabled = true
-  end
-
+    for display,displaydata in pairs(DID.displays) do
+        if displaydata.IR_unlock and force.technologies[displaydata.IR_unlock] and force.technologies[displaydata.IR_unlock].researched then
+            force.recipes[display].enabled = true
+        elseif displaydata.unlock and force.technologies[displaydata.unlock] and force.technologies[displaydata.unlock].researched then
+            force.recipes[display].enabled = true
+        end
+    end
 end