diff --git a/__init__.py b/__init__.py
index e2d4076..28aa6bd 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from .color_transfer import PaletteTransferNode, PalleteTransferClustering, PaletteTransferReinhard, PaletteSoftTransfer, PaletteRbfTransfer, PaletteOptimalTransportTransfer, ReferenceTransferReinhard, ColorPaletteNode
+from .color_transfer import PaletteTransferNode, PalleteTransferClustering, PaletteTransferReinhard, PaletteSoftTransfer, PaletteRbfTransfer, PaletteOptimalTransportTransfer, ReferenceTransferReinhard, ColorPaletteNode, ExtractPaletteNode
 
 
 NODE_CLASS_MAPPINGS = {
@@ -10,6 +10,7 @@ NODE_CLASS_MAPPINGS = {
     "PaletteOptimalTransportTransfer": PaletteOptimalTransportTransfer,
     "ColorTransferReinhard": ReferenceTransferReinhard,
     "ColorPalette": ColorPaletteNode,
+    "ExtractPalette": ExtractPaletteNode
 }
 NODE_DISPLAY_NAME_MAPPINGS = {
     "PaletteTransfer": "Palette Transfer",