Update TrafficShapingTest.java