Add support for afterAll