Improved drive quantity calculation