Fixed a small bug in main.js.