Fixed issue with regressor init