mocha test fixes