Tidy up

diff --git a/tests/test_pgwui_server.py b/tests/test_pgwui_server.py
index e8532b7150b927c058a6dfe3d34f571cdcd96760..ac449215c7f493ddcd7815acace7c543f0405bcd 100644 (file)
--- a/tests/test_pgwui_server.py
+++ b/tests/test_pgwui_server.py
@@ -88,7 +88,7 @@ def test_dot_to_multiline_setting_new(mock_parse_assignments):
 
     mock_parse_assignments.return_value = comp_settings
     pgwui_server.dot_to_multiline_setting(
-        settings, key, 'pgwui_component')
+        settings, key, component)
 
     assert settings == expected
 
@@ -106,7 +106,7 @@ def test_dot_to_multiline_setting_old(mock_parse_assignments):
 
     mock_parse_assignments.return_value = comp_settings
     pgwui_server.dot_to_multiline_setting(
-        settings, key, 'pgwui_component')
+        settings, key, component)
 
     assert settings == expected