diff --git a/tests/test_inference.py b/tests/test_inference.py
index 303aa67..5eb7442 100644
--- a/tests/test_inference.py
+++ b/tests/test_inference.py
@@ -1,6 +1,5 @@
 import pytest
 import torch
-import numpy as np
 
 from enhancer.inference import Inference