Properly update data partition size