diff --git a/assets/src/scripts/components/hydrograph/legend.js b/assets/src/scripts/components/hydrograph/legend.js
index 3596f0b5c098dbcbd9213ae74ccf78ee384d875f..59c0c224262793e21d940ec1409ff998b489965f 100644
--- a/assets/src/scripts/components/hydrograph/legend.js
+++ b/assets/src/scripts/components/hydrograph/legend.js
@@ -156,7 +156,7 @@ export const drawSimpleLegend = function(div, {legendMarkerRows, layout}) {
                 xPosition = markerGroupBBox.x + markerGroupBBox.width + markerGroupXOffset;
 
             } catch(error) {
-                null;
+                // See above explanation
             }
         });
     });
@@ -176,15 +176,9 @@ const uniqueClassesSelector = memoize(tsKey => createSelector(
     (tsLineSegments) => {
         let classes = [].concat(...Object.values(tsLineSegments)).map((line) => line.classes);
         return {
-            default: classes.find((cls) => {
-               return !cls.approved && !cls.estimated && !cls.dataMask;
-            }) ? true : false,
-            approved: classes.find((cls) => {
-                return cls.approved;
-            })? true : false,
-            estimated: classes.find((cls) => {
-                return cls.estimated;
-            }) ? true : false,
+            default: classes.some((cls) => !cls.approved && !cls.estimated && !cls.dataMask),
+            approved: classes.some((cls) => cls.approved),
+            estimated: classes.some((cls) => cls.estimated),
             dataMasks: new Set(classes.map((cls) => cls.dataMask).filter((mask) => {
                 return mask;
             }))