fix progress calculation while draining