fix main.lua