Make prediction work with unsolo command