Make compatible with mocha 2.x

See #6