Get the partition tests working