diff --git a/tests/core/test_rai_cli.py b/tests/core/test_rai_cli.py
index 98e370a25..be49b0703 100644
--- a/tests/core/test_rai_cli.py
+++ b/tests/core/test_rai_cli.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import os
-import uuid
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 
diff --git a/tests/messages/test_utils.py b/tests/messages/test_utils.py
index 4f746a127..b537f91bb 100644
--- a/tests/messages/test_utils.py
+++ b/tests/messages/test_utils.py
@@ -16,6 +16,7 @@
 from io import BytesIO
 
 import numpy as np
+import pytest
 from PIL import Image
 
 from rai.messages.utils import preprocess_image
@@ -38,4 +39,3 @@ def decode_image(base64_string: str) -> Image.Image:
 def test_preprocess_image(test_image):
     base64_image = preprocess_image(test_image)
     _ = decode_image(base64_image)  # noqa: F841
-