diff --git a/assets/src/scripts/components/hydrograph/index.spec.js b/assets/src/scripts/components/hydrograph/index.spec.js
index b0615342c43d6a8639276a664ab539a4d1161c81..d5205208f9468a3e006f5b6f8a33c9ac1553c515 100644
--- a/assets/src/scripts/components/hydrograph/index.spec.js
+++ b/assets/src/scripts/components/hydrograph/index.spec.js
@@ -151,16 +151,11 @@ const TEST_STATE = {
     },
     floodData: {
         floodLevels: {
-            sites:
-                [
-                    {
-                        site_no: '07144100',
-                        action_stage: '20',
-                        flood_stage: '22',
-                        moderate_flood_stage: '25',
-                        major_flood_stage: '26'
-                    }
-                ]
+            site_no: '07144100',
+            action_stage: '20',
+            flood_stage: '22',
+            moderate_flood_stage: '25',
+            major_flood_stage: '26'
         }
     }
 };
diff --git a/assets/src/scripts/components/hydrograph/legend.spec.js b/assets/src/scripts/components/hydrograph/legend.spec.js
index 4bcdfe086e08322ef5f97dac2e6e8ca5f26b2830..c17b6349cc96660ec01492a3edd33ef2b2344a3b 100644
--- a/assets/src/scripts/components/hydrograph/legend.spec.js
+++ b/assets/src/scripts/components/hydrograph/legend.spec.js
@@ -97,16 +97,11 @@ describe('UV: Legend module', () => {
         },
         floodData: {
             floodLevels: {
-                sites:
-                    [
-                        {
-                            site_no: '07144100',
-                            action_stage: '20',
-                            flood_stage: '22',
-                            moderate_flood_stage: '25',
-                            major_flood_stage: '26'
-                        }
-                    ]
+                site_no: '07144100',
+                action_stage: '20',
+                flood_stage: '22',
+                moderate_flood_stage: '25',
+                major_flood_stage: '26'
             }
         }
     };
diff --git a/assets/src/scripts/components/hydrograph/time-series-graph.spec.js b/assets/src/scripts/components/hydrograph/time-series-graph.spec.js
index 3b13a2032dc733d56df53ae12d0efe3a2fe18a3d..af75275ea65f90ddfe8faf3e916cf12fec8a69be 100644
--- a/assets/src/scripts/components/hydrograph/time-series-graph.spec.js
+++ b/assets/src/scripts/components/hydrograph/time-series-graph.spec.js
@@ -180,16 +180,11 @@ const TEST_STATE = {
     },
     floodData: {
         floodLevels: {
-            sites:
-                [
-                    {
-                        site_no: '07144100',
-                        action_stage: '20',
-                        flood_stage: '22',
-                        moderate_flood_stage: '25',
-                        major_flood_stage: '26'
-                    }
-                ]
+            site_no: '07144100',
+            action_stage: '20',
+            flood_stage: '22',
+            moderate_flood_stage: '25',
+            major_flood_stage: '26'
         }
     }
 };
diff --git a/assets/src/scripts/selectors/flood-data-selector.spec.js b/assets/src/scripts/selectors/flood-data-selector.spec.js
index d7c0b9d87758f4d26480367d0fa239067300a10e..433e0271c06343f49507214798152964c1c3d6e5 100644
--- a/assets/src/scripts/selectors/flood-data-selector.spec.js
+++ b/assets/src/scripts/selectors/flood-data-selector.spec.js
@@ -164,20 +164,15 @@ describe('flood-data-selector', () => {
 
          it('Return true if waterwatch flood levels should be visible', () =>{
             expect(waterwatchVisible({
-                floodData: {
-                    floodLevels: {
-                        sites:
-                            [
-                                {
-                                    site_no: '07144100',
-                                    action_stage: '20',
-                                    flood_stage: '22',
-                                    moderate_flood_stage: '25',
-                                    major_flood_stage: '26'
-                                }
-                            ]
-                    }
-                },
+                 floodData: {
+                     floodLevels: {
+                         site_no: '07144100',
+                         action_stage: '20',
+                         flood_stage: '22',
+                         moderate_flood_stage: '25',
+                         major_flood_stage: '26'
+                     }
+                 },
                 ivTimeSeriesState: {
                     currentIVVariableID: '45807197'
                 },