Update main.js

fix bug