diff --git a/tests/test_rl/test_policy_store.py b/tests/test_rl/test_policy_store.py
index 2f4ace5..5dea003 100644
--- a/tests/test_rl/test_policy_store.py
+++ b/tests/test_rl/test_policy_store.py
@@ -2,8 +2,6 @@
 from __future__ import annotations
 
 import json
-import tempfile
-from pathlib import Path
 
 import numpy as np
 import pytest
diff --git a/tests/test_rl/test_simulator.py b/tests/test_rl/test_simulator.py
index a5735bd..102972d 100644
--- a/tests/test_rl/test_simulator.py
+++ b/tests/test_rl/test_simulator.py
@@ -2,10 +2,9 @@
 from __future__ import annotations
 
 import numpy as np
-import pytest
 
 from saar.rl.action_space import N_ACTIONS
-from saar.rl.simulator import Episode, SaarSimulator
+from saar.rl.simulator import SaarSimulator
 
 
 class TestSaarSimulator: