Actually fixed the regression