Fixed issue with regressor type handling