Update main.js

bug fix