Update for testling with mocha