Updated mocha tests