Fix main.js bug