diff --git a/label_studio_ml/response.py b/label_studio_ml/response.py
index 546be526b..29f36162a 100644
--- a/label_studio_ml/response.py
+++ b/label_studio_ml/response.py
@@ -1,8 +1,8 @@
 
-from typing import Type, Dict, Optional, List, Tuple, Any, Union
-from pydantic import BaseModel, confloat
+from typing import Optional, List
+from pydantic import BaseModel
 
-from label_studio_sdk.objects import PredictionValue
+from label_studio_sdk._legacy.objects import PredictionValue
 
 
 class ModelResponse(BaseModel):