small edit to main.js