diff --git a/tests/test_label_studio_skill.py b/tests/test_label_studio_skill.py
index 5288d36..8ef1766 100644
--- a/tests/test_label_studio_skill.py
+++ b/tests/test_label_studio_skill.py
@@ -2,6 +2,8 @@
 import os
 import pandas as pd
 from adala.agents import Agent
+from label_studio_sdk.label_interface import LabelInterface
+from label_studio_sdk.label_interface.objects import PredictionValue
 
 
 @pytest.mark.asyncio
@@ -161,10 +163,6 @@ async def test_label_studio_skill_with_ner():
     assert predictions.entities.tolist() == expected_predictions
 
 
-from label_studio_sdk.label_interface import LabelInterface
-from label_studio_sdk.label_interface.objects import PredictionValue
-
-
 @pytest.mark.vcr
 @pytest.mark.asyncio
 async def test_label_studio_skill_valid_predictions():