Fix referee.py