Fixed unit test for main.js