Fix disk size calculation