fix bug in main.js
fix it