Small change to unify the two main.js