Update test_companies.py