diff --git a/pylossless/dash/tests/test_topo_viz.py b/pylossless/dash/tests/test_topo_viz.py
index 1cf4562..237f5f2 100644
--- a/pylossless/dash/tests/test_topo_viz.py
+++ b/pylossless/dash/tests/test_topo_viz.py
@@ -52,7 +52,7 @@ def test_GridTopoPlot():
     offset = 2
     nb_topo = 4
     plot_data = topo_data.topo_values.iloc[::-1].iloc[offset : offset + nb_topo]
-    plot_data = plot_data.values.tolist()
+    plot_data = list(plot_data.T.to_dict().values())
 
     GridTopoPlot(
         2,