Remove ensemble dependency