Fix in partition key comparison