Regression fixes