fixing feature to add new unit thumbnail behavior