Fix main.lua