Fix partition count == 1