Make invocation after end() expect a callback, to play nice with e.g. mocha