Fix regression in AfterAll