diff --git a/wooden-arig/migrations/1.0.0.lua b/wooden-arig/migrations/1.0.1.lua
similarity index 64%
rename from wooden-arig/migrations/1.0.0.lua
rename to wooden-arig/migrations/1.0.1.lua
index 4b5f70f..53b0b3a 100644
--- a/wooden-arig/migrations/1.0.0.lua
+++ b/wooden-arig/migrations/1.0.1.lua
@@ -1,4 +1,4 @@
-local arig = game.get_surface("arig")
+local arig = game.get_surface("arig") or game.planets["arig"].create_surface()
 
 local mgs = arig.map_gen_settings
 mgs.autoplace_settings.decorative.settings["arig-small-cactus"] = nil