Fixed shrink in partition_module. Must test it