Fixed issue in main.js