Fix incorrect import in test_match.py