Fixed broken partitioning test

partitioning was broken, now it’s fixed.