small fix main.js