diff --git a/gauge/sdk/tests/test_sdk.py b/gauge/sdk/tests/test_sdk.py
index 255327a..54f3dd9 100644
--- a/gauge/sdk/tests/test_sdk.py
+++ b/gauge/sdk/tests/test_sdk.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from gauge.sdk.main import endpoint
+from gauge import endpoint
 
 
 def test_default_endpoint_decorator():