Fix a bug in partition.

I botched that last one. Must mind my 1's and i's.