Probably not. If it's a bug then it will be addressed eventually, but if intentional then I expect we're stuck with it. Think of it this way - previously you probably were not really getting a full charge. If you take it off the charger while it is doing that last long part of the charge, the resulting longevity may be the same as before - if as I speculate, it was falsely and temporarily registering full charge. I recall a number of us posting about how under Lollipop we'd take it off the charger and it would quickly drop. Mine no longer seems to do that. It would be great to get confirmation on this but doubt we'll get it. I can live with it if it gives me a better charge.
BTW - note that for the new doze feature to work, you need to sit the phone down when you're not using it, since it uses the accelerometer/gyro sensor to determine motion. So - if you carry your phone in a belt holster or pocket, then even subtle motion will prevent doze mode from engaging.