Set proper partition sizes