fixing and issue in the tests (giving beck the partition when still changing it)