fixed issue with ensemble