diff --git a/src/components/ChartNode.js b/src/components/ChartNode.js
index a2f1a47..f938963 100644
--- a/src/components/ChartNode.js
+++ b/src/components/ChartNode.js
@@ -40,6 +40,7 @@ const ChartNode = ({
 
   const nodeClass = [
     "oc-node",
+    datasource.className,
     isChildrenCollapsed ? "isChildrenCollapsed" : "",
     allowedDrop ? "allowedDrop" : "",
     selected ? "selected" : ""