diff --git a/tests/test_sim.py b/tests/test_sim.py
index 7d3c087..cc475e8 100644
--- a/tests/test_sim.py
+++ b/tests/test_sim.py
@@ -3,6 +3,7 @@ from __future__ import annotations
 
 import numpy as np
 import pandas as pd
+import pytest
 
 from pluginbattery.sim import (
     Battery,