diff --git a/mindflow/__init__.py b/mindflow/__init__.py
index 6a9beea..3d18726 100644
--- a/mindflow/__init__.py
+++ b/mindflow/__init__.py
@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.5.0"
diff --git a/mindflow/core/types/store_traits/json.py b/mindflow/core/types/store_traits/json.py
index 3e1472a..28cbb13 100644
--- a/mindflow/core/types/store_traits/json.py
+++ b/mindflow/core/types/store_traits/json.py
@@ -92,7 +92,7 @@ def delete_bulk(cls, object_ids: List[str]):
             objects.pop(object_id, None)
 
     def save(self):
-        if not (objects := collections.get(self.__name__, {})):
+        if not (objects := collections.get(self.__class__.__name__, {})):
             collections[self.__class__.__name__] = objects
 
         if not hasattr(self, "id"):