Fix partition sizes