diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index 3cff98d701ea06eaf25b8d988775506b02ff11ed..d213e8bf8f358b98264e1046fc25c0ffc0f04c61 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -100,7 +100,7 @@ class DAN:
         :param confidences: Return the characters probabilities.
         :param attentions: Return characters attention weights.
         """
-        input_tensor.to(self.device)
+        input_tensor = input_tensor.to(self.device)
 
         start_token = len(self.charset) + 1
         end_token = len(self.charset)