Reassigning of assigned vehicle in the end of its upgrading replacement.

I noticed that replacing the assigned vehicle did not affect the tree harvester to which the replaced vehicle was assigned. The same type of vehicle just drove up to the tree harvester.
Please, make option for replacing assigned vehicle to reassign upgraded vehicle to same mine tower or harvester or etc. With warning if upgraded vehicle will be incompatible with reassignment and will not be assigned to same thing.