diff --git a/factories/DisplayFactory.py b/factories/DisplayFactory.py
index c140238..a55c2f3 100644
--- a/factories/DisplayFactory.py
+++ b/factories/DisplayFactory.py
@@ -6,9 +6,9 @@ class DisplayFactory:
         self.displayNames = displayNames
 
     def createSpecialDisplay(self):
-        name = next(displayName for displayName in self.displayNames if "Special" in displayName)
+        name = next(displayName.name for displayName in self.displayNames if "Special" in displayName.name)
         return SpecialDisplay(name)
 
     def createPlayerDisplay(self, id):
-        name = next(displayName for displayName in self.displayNames if "Player" in displayName and str(id) in displayName)
+        name = next(displayName.name for displayName in self.displayNames if "Player" in displayName.name and str(id) in displayName.name)
         return PlayerDisplay(name)