update integration test to work with recent changes