bug fix

bug fix on main.js