Not all tests do this the same way and I was trying to minimise changes where possible.