modified remove_mono for prediction phase