Updated the mocha tests