diff --git a/jsonschema_gentypes/__init__.py b/jsonschema_gentypes/__init__.py
index 7ec04fae..ba6459bb 100644
--- a/jsonschema_gentypes/__init__.py
+++ b/jsonschema_gentypes/__init__.py
@@ -730,10 +730,12 @@ def get_name(
         output += postfix
     if not get_name.__dict__.get("names"):
         get_name.__dict__["names"] = set()
+        # get_name.__dict__["names"] = dict()
     # elif output in get_name.__dict__["names"] and get_name.__dict__["names"][output] != schema:
     elif output in get_name.__dict__["names"]:
         output += str(random.randint(0, 9999))
-    get_name.__dict__["names"][output] = schema
+    # get_name.__dict__["names"][output] = schema
+    get_name.__dict__["names"].add(output)
     return output