Added a fix for failing partition-table test