Fixed regression

Great, thanks