Fixed error in main.lua