Make unit test compatible with mocha async-only