diff --git a/tests/test_cli.py b/tests/test_cli.py
index 4cbf1209..3c99f424 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -2,9 +2,6 @@
 from pathlib import Path
 from random import random
 
-# External Modules
-import pytest
-
 # Internal Packages
 from src.utils.cli import cli
 from src.utils.helpers import resolve_absolute_path
diff --git a/tests/test_image_search.py b/tests/test_image_search.py
index 80c4fdf6..ad374da1 100644
--- a/tests/test_image_search.py
+++ b/tests/test_image_search.py
@@ -48,8 +48,13 @@ def test_image_search(content_config: ContentConfig, search_config: SearchConfig
             image_files_url='/static/images',
             count=1)
 
-        actual_image = Image.open(output_directory.joinpath(Path(results[0]["entry"]).name))
+        actual_image_path = output_directory.joinpath(Path(results[0]["entry"]).name)
+        actual_image = Image.open(actual_image_path)
         expected_image = Image.open(content_config.image.input_directories[0].joinpath(expected_image_name))
 
         # Assert
         assert expected_image == actual_image
+
+    # Cleanup
+    # Delete the image files copied to results directory
+    actual_image_path.unlink()
diff --git a/tests/test_org_to_jsonl.py b/tests/test_org_to_jsonl.py
index f9539b26..39b3e91f 100644
--- a/tests/test_org_to_jsonl.py
+++ b/tests/test_org_to_jsonl.py
@@ -1,6 +1,5 @@
 # Standard Packages
 import json
-from posixpath import split
 
 # Internal Packages
 from src.processor.org_mode.org_to_jsonl import convert_org_entries_to_jsonl, extract_org_entries