Update TracingStageTest