diff --git a/parts/modes/custom_puzzle.lua b/parts/modes/custom_puzzle.lua
index faae7e36..278bc245 100644
--- a/parts/modes/custom_puzzle.lua
+++ b/parts/modes/custom_puzzle.lua
@@ -40,6 +40,7 @@ return{
 	load=function()
 		applyCustomGame()
 		PLY.newPlayer(1)
+		local ENV=GAME.modeEnv
 		local AItype=ENV.opponent:sub(1,2)
 		local AIlevel=tonumber(ENV.opponent:sub(-1))
 		if AItype=="9S"then