diff --git a/tests/test_bleu_tchmet.py b/tests/test_bleu_tchmet.py
index d6c8537..13563e6 100644
--- a/tests/test_bleu_tchmet.py
+++ b/tests/test_bleu_tchmet.py
@@ -2,9 +2,9 @@
 # -*- coding: utf-8 -*-
 
 import unittest
-
 from unittest import TestCase
 
+from torch import Tensor
 from torchmetrics.text.bleu import BLEUScore
 
 from aac_metrics.classes.bleu import BLEU