Fix integration tests to fit new result format