Small change to aggregator.py