Update mocha tests