Adapt test to transport changes