diff --git a/graphiti_core/utils/maintenance/node_operations.py b/graphiti_core/utils/maintenance/node_operations.py
index a84b8b94..751ba3ac 100644
--- a/graphiti_core/utils/maintenance/node_operations.py
+++ b/graphiti_core/utils/maintenance/node_operations.py
@@ -189,6 +189,9 @@ async def extract_nodes(
     new_nodes = []
     for name in extracted_node_names:
         entity_type = node_classifications.get(name)
+        if entity_types is not None and entity_type not in entity_types:
+            entity_type = None
+
         labels = (
             ['Entity']
             if entity_type is None or entity_type == 'None' or entity_type == 'null'
diff --git a/pyproject.toml b/pyproject.toml
index 233692ac..e143cf9c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphiti-core"
-version = "0.8.0"
+version = "0.8.1"
 description = "A temporal graph building library"
 authors = [
     "Paul Paliychuk <paul@getzep.com>",