Update main.js

Fixed a bug.