diff --git a/tests/unit_tests/weightwindows/test_wwinp_reader.py b/tests/unit_tests/weightwindows/test_wwinp_reader.py
index 437e588f66..8d7cd5f7d2 100644
--- a/tests/unit_tests/weightwindows/test_wwinp_reader.py
+++ b/tests/unit_tests/weightwindows/test_wwinp_reader.py
@@ -118,12 +118,7 @@ def test_wwinp_reader(wwinp_data, request):
         # check the expected weight window values mocked in the file --
         # a reversed array of the flat index into the numpy array
         n_wws = np.prod((*mesh.dimension, e_bounds.size - 1))
-        nx, ny, nz, ne = mesh.dimension + (e_bounds.size - 1,)
-        exp_ww_lb = np.linspace(1, n_wws, n_wws)[::-1].reshape(nx, ny, nz, ne)
-        # this is how the ww lower bounds were actually written to file
-        exp_ww_lb = np.swapaxes(exp_ww_lb, 0, 2).flatten()
-        # this is how we expect them to be read back in
-        exp_ww_lb = exp_ww_lb.reshape(ne, nz, ny, nx).T.flatten()
+        exp_ww_lb = np.linspace(1, n_wws, n_wws)[::-1]
 
         np.testing.assert_array_equal(exp_ww_lb, ww.lower_ww_bounds.flatten())