Fix highlights.py