fix ensembling
fix tests