Fix use of mocha gem