Update test_related.py