diff --git a/test/unit/specs/vislib/lib/legend.js b/test/unit/specs/vislib/lib/legend.js
index 01d8fe8eec9e4..a8dfa16538ed9 100644
--- a/test/unit/specs/vislib/lib/legend.js
+++ b/test/unit/specs/vislib/lib/legend.js
@@ -70,7 +70,7 @@ define(function (require) {
               return path.getAttribute('data-label');
             })
             .filter(function (dataLabel) {
-              return dataLabel === label;
+              return dataLabel === label + '';
             })
             .value();