diff --git a/factories/DisplayFactory.py b/factories/DisplayFactory.py index fb14e4f..f2764b2 100644 --- a/factories/DisplayFactory.py +++ b/factories/DisplayFactory.py @@ -8,11 +8,10 @@ class DisplayFactory: def createDisplays(self): displays = [None] * config.MAX_PLAYERS - for displayName in self.displayNames: - if "Player" in displayName: - id = displayName.replace("Player", "") - displays[id] = PlayerDisplay(displayName) - print(displays) + for display in self.displayNames: + if "Player" in display.name: + id = int(display.name.replace("Player", "")) + displays[id] = PlayerDisplay(display.name) return displays def createSpecialDisplay(self):