Update main.lua

Fixed another bug