diff --git a/tests/test_backends_torch_gradients.py b/tests/test_backends_torch_gradients.py
index 3ea2a6ee03..abb6339ed5 100644
--- a/tests/test_backends_torch_gradients.py
+++ b/tests/test_backends_torch_gradients.py
@@ -5,7 +5,7 @@
 from qiboml.backends import PyTorchBackend
 
 from qibo import Circuit, gates
-from qibo.backends import PyTorchBackend, construct_backend
+from qibo.backends import construct_backend
 from qibo.quantum_info.metrics import infidelity