Fix cluster.py script.