Fixed the assert_equal and partition table issue