fix test

SimonBlanke · SimonBlanke · commit 1af2dad444e4 · 2026-04-12T13:54:25.000+02:00
diff --git a/tests/core/custom_test_function/test_storage.py b/tests/core/custom_test_function/test_storage.py
@@ -13,8 +13,8 @@
 
 import pytest
 
-from surfaces.custom_test_function import (
-    CustomTestFunction,
+from surfaces import CustomTestFunction
+from surfaces.test_functions._custom_storage import (
     InMemoryStorage,
     SQLiteStorage,
     Storage,
