A small thing, but can you move this to before the asserts so it gets called even if the test fails?