diff --git a/tests/test_visualizations.py b/tests/test_visualization.py
similarity index 97%
rename from tests/test_visualizations.py
rename to tests/test_visualization.py
index ac02cee..8505b61 100644
--- a/tests/test_visualizations.py
+++ b/tests/test_visualization.py
@@ -4,7 +4,7 @@ import graphviz
 import pytest
 
 from agents import Agent
-from agents.extensions.visualizations import draw_graph, get_all_edges, get_all_nodes, get_main_graph
+from agents.extensions.visualization import draw_graph, get_all_edges, get_all_nodes, get_main_graph
 
 
 @pytest.fixture