fixed integration test to work with new schema