diff --git a/src/plette/__init__.py b/src/plette/__init__.py
index 3f38095..ec56d5d 100644
--- a/src/plette/__init__.py
+++ b/src/plette/__init__.py
@@ -3,7 +3,7 @@
     "Lockfile", "Pipfile",
 ]
 
-__version__ = '2.0.0'
+__version__ = '2.0.1'
 
 from .lockfiles import Lockfile
 from .pipfiles import Pipfile
diff --git a/src/plette/lockfiles.py b/src/plette/lockfiles.py
index b04d67b..fcb8779 100644
--- a/src/plette/lockfiles.py
+++ b/src/plette/lockfiles.py
@@ -115,7 +115,7 @@ def __getitem__(self, key):
             return value
 
     def __setitem__(self, key, value):
-        if isinstance(value, DataView):
+        if isinstance(value, DataModel):
             self._data[key] = value._data
         else:
             self._data[key] = value