Update to mocha test