diff --git a/speechbrain/alignment/aligner.py b/speechbrain/alignment/aligner.py
index bb55e5b469..020dd851b1 100644
--- a/speechbrain/alignment/aligner.py
+++ b/speechbrain/alignment/aligner.py
@@ -739,7 +739,7 @@ def _dp_viterbi(
             )
             v_matrix[:, :, t] = x + emiss_pred_useful[:, :, t]
 
-            backpointers[:, :, t] = argmax.type(torch.FloatTensor)
+            backpointers[:, :, t] = argmax.type(dtype=torch.float32)
 
         z_stars = []
         z_stars_loc = []