diff --git a/src/components/shapes/draw.js b/src/components/shapes/draw.js
index 556920c80a9..66ffe569d74 100644
--- a/src/components/shapes/draw.js
+++ b/src/components/shapes/draw.js
@@ -93,7 +93,8 @@ function drawOne(gd, index) {
         var attrs = {
             'data-index': index,
             'fill-rule': 'evenodd',
-            d: getPathString(gd, options)
+            d: getPathString(gd, options),
+            class: options._input.className
         };
         var lineColor = options.line.width ? options.line.color : 'rgba(0,0,0,0)';