Fix main memory in get-baseline.py.