Integration test tweak