diff --git a/__init__.py b/__init__.py
index 215ce66..6866e91 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,11 +1,13 @@
-from .bria_api_node import EraserNode
+from .bria_api_node import EraserNode, GenerativeFillNode
 
 # Map the node class to a name used internally by ComfyUI
 NODE_CLASS_MAPPINGS = {
     "BriaEraser": EraserNode,  # Return the class, not an instance
+    "BriaGenerativeFill": GenerativeFillNode,
 }
 
 # Map the node display name to the one shown in the ComfyUI node interface
 NODE_DISPLAY_NAME_MAPPINGS = {
     "BriaEraser": "Bria Eraser",
+    "BriaGenerativeFill": "Bria Generative Fill",
 }