Make test setup code a bit prettier