implement setPredictType correctly for wrappers