diff --git a/tests/test_utils_snapshot.py b/tests/test_utils_snapshot.py
index 6f3b1d214..3b75e16a9 100644
--- a/tests/test_utils_snapshot.py
+++ b/tests/test_utils_snapshot.py
@@ -1,8 +1,5 @@
 import pytest
-import sys
-import os
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
-from slic.utils.snapshot import *
+from slic.utils import snapshot 
 from unittest.mock import patch
 
 
@@ -34,7 +31,7 @@ def test_snapshot_excludes_internal_by_default():
     hidden = FakeAdjustable("internal", internal=True)
 
     with patch("slic.utils.snapshot.instances", return_value=[visible, hidden]):
-        result = snapshot()
+        result = snapshot.snapshot()
         assert visible in result
         assert hidden not in result
 
@@ -45,7 +42,7 @@ def test_snapshot_include_internal_flag():
     hidden = FakeAdjustable("internal", internal=True)
 
     with patch("slic.utils.snapshot.instances", return_value=[visible, hidden]):
-        result = snapshot(include_internal=True)
+        result = snapshot.snapshot(include_internal=True)
         assert visible in result
         assert hidden in result
         assert len(result) == 2
@@ -54,7 +51,7 @@ def test_snapshot_include_internal_flag():
 def test_snapshot_empty_result():
 
     with patch("slic.utils.snapshot.instances", return_value=[]):
-        result = snapshot()
+        result = snapshot.snapshot()
         assert isinstance(result, list)
         assert result == []
 
@@ -84,7 +81,7 @@ def test_snapshot_sort_keys(sort_key, expected_order):
     unordered = [c, a, b]  # intentionally unsorted
 
     with patch("slic.utils.snapshot.instances", return_value=unordered):
-        result = snapshot(sort_key=sort_key)
+        result = snapshot.snapshot(sort_key=sort_key)
         # We check that the objects returned match the expected names (using original mapping)
         result_names = [adj.name for adj in result]
         assert result_names == [name_to_obj[n].name for n in expected_order]