diff --git a/tests/test_graph.py b/tests/test_clip.py
similarity index 89%
rename from tests/test_graph.py
rename to tests/test_clip.py
index 4ca84ce..d67cf92 100644
--- a/tests/test_graph.py
+++ b/tests/test_clip.py
@@ -1,8 +1,8 @@
 import pytest
-from package.graph import display
+from package.clip import display
 from datetime import datetime
 import subprocess
-from package.graph import copy_to_clipboard
+from package.clip import copy_to_clipboard
 import os
 import sys
 import shutil
diff --git a/tests/test_models.py b/tests/test_show.py
similarity index 88%
rename from tests/test_models.py
rename to tests/test_show.py
index 1912666..1273684 100644
--- a/tests/test_models.py
+++ b/tests/test_show.py
@@ -1,7 +1,7 @@
 import pytest
 import sys
 from datetime import datetime
-from package.models import display, copy_to_clipboard
+from package.show import display, copy_to_clipboard
 
 # Mock datetime class
 class MockDateTime(datetime):
@@ -25,7 +25,7 @@ def read(self):
 
 
 def test_display_incorrect_password(monkeypatch):
-    monkeypatch.setattr("package.piechart.datetime", MockDateTime)  # Mock datetime
+    monkeypatch.setattr("package.show.datetime", MockDateTime)  # Mock datetime
 
     snippet_name = "test"
     incorrect_password = "1111"  # Different from "1234"
diff --git a/tests/test_piechart.py b/tests/test_write.py
similarity index 85%
rename from tests/test_piechart.py
rename to tests/test_write.py
index e99b029..cb7eb7d 100644
--- a/tests/test_piechart.py
+++ b/tests/test_write.py
@@ -1,7 +1,7 @@
 import shutil
 import glob
 from datetime import datetime
-from package.piechart import plot  
+from package.write import plot  
 
 #we cannot mock this function, so mocking the whole class
 class MockDateTime(datetime):
@@ -22,7 +22,7 @@ def mock_glob_multiple(pattern):
     return ["stash/test.py", "stash/test2.py"]
 
 def test_plot_success(monkeypatch, capfd):
-    monkeypatch.setattr("package.piechart.datetime", MockDateTime)
+    monkeypatch.setattr("package.write.datetime", MockDateTime)
     monkeypatch.setattr(shutil, "copyfile", mock_copyfile)
     monkeypatch.setattr(glob, "glob", mock_glob_single)
     
@@ -36,7 +36,7 @@ def test_plot_success(monkeypatch, capfd):
 
 
 def test_plot_no_file_found(monkeypatch, capfd):
-    monkeypatch.setattr("package.piechart.datetime", MockDateTime)
+    monkeypatch.setattr("package.write.datetime", MockDateTime)
     monkeypatch.setattr(glob, "glob", mock_glob_none)
     
     snippet_name = "nonexistent"
@@ -48,7 +48,7 @@ def test_plot_no_file_found(monkeypatch, capfd):
     assert "File is not found" in captured.out
 
 def test_plot_multiple_files_found(monkeypatch, capfd):
-    monkeypatch.setattr("package.piechart.datetime", MockDateTime)
+    monkeypatch.setattr("package.write.datetime", MockDateTime)
     monkeypatch.setattr(glob, "glob", mock_glob_multiple)
     
     snippet_name = "test"