Minor fix in main.js