Update main.js

fix