add_partition_check now is working and tested