diff --git a/ultralytics/models/fastsam/prompt.py b/ultralytics/models/fastsam/prompt.py
index 72ea5ef864..0f43441a5c 100644
--- a/ultralytics/models/fastsam/prompt.py
+++ b/ultralytics/models/fastsam/prompt.py
@@ -342,7 +342,7 @@ class FastSAMPrompt:
             max_idx = scores.argsort()
             max_idx = max_idx[-1]
             max_idx += sum(np.array(filter_id) <= int(max_idx))
-            self.results[0].masks.data = torch.tensor(np.array([ann['segmentation'] for ann in annotations]))
+            self.results[0].masks.data = torch.tensor(np.array([annotations[max_idx]['segmentation']]))
         return self.results
 
     def everything_prompt(self):