Need to clean current_model along with current_shard