Fix the mocha issue