Temporary hack to make ensemble work properly.