Update main.js

Minor fix.