Change return value to be array instead of dict. aggregate.py creates dict.