diff --git a/stumpy/stump.py b/stumpy/stump.py
index dba9abcf5..f77d6601e 100644
--- a/stumpy/stump.py
+++ b/stumpy/stump.py
@@ -161,7 +161,7 @@ def _compute_diagonal(
     n_B = T_B.shape[0]
     m_inverse = 1.0 / m
     constant = (m - 1) * m_inverse * m_inverse  # (m - 1)/(m * m)
-    uint64_m = np.uint64(m)
+    uint64_m = m # np.uint64(m)
 
     for diag_idx in range(diags_start_idx, diags_stop_idx):
         g = diags[diag_idx]
@@ -172,8 +172,8 @@ def _compute_diagonal(
             iter_range = range(-g, min(n_A - m + 1, n_B - m + 1 - g))
 
         for i in iter_range:
-            uint64_i = np.uint64(i)
-            uint64_j = np.uint64(i + g)
+            uint64_i = i # np.uint64(i)
+            uint64_j = j # np.uint64(i + g)
 
             if uint64_i == 0 or uint64_j == 0:
                 cov = (