Update 002_latency_test.py