Update mocha unit test