Fixing mocha test