Fixes to regression.py