Small correction in size calculation