Update law_firms_test.py