Fixed a bug in getting disk size from partition