Improve mocha setup, still one test failure