fix main.js bug