Assign drive correctly on update drive.