Update main.lua

Should fix #37, thanks @boolsheet for the testing.