Update test_flight.py