Update integration_test.py