Update test_integration.py