diff --git a/neo/pt.py b/neo/pt.py index 7f97e9f78e64bf5340f52f7ed5f09340a174203b..cc30476bf70f5e90c1dffedaa3072d37756204e0 100644 --- a/neo/pt.py +++ b/neo/pt.py @@ -186,13 +186,13 @@ class PartitionTable(object): def removeCell(self, offset, node): row = self.partition_list[offset] - if row is not None: - for cell in row: - if cell.getNode() == node: - row.remove(cell) - if not cell.isFeeding(): - self.count_dict[node] -= 1 - break + assert row is not None + for cell in row: + if cell.getNode() == node: + row.remove(cell) + if not cell.isFeeding(): + self.count_dict[node] -= 1 + break def load(self, ptid, row_list, nm): """