Fixed broken mocha tests.