Wrap afterEach in AIT.wrap