diff --git a/stemflow/utils/plot_gif.py b/stemflow/utils/plot_gif.py
index 09d6bfc..13a3d20 100644
--- a/stemflow/utils/plot_gif.py
+++ b/stemflow/utils/plot_gif.py
@@ -147,10 +147,10 @@ def animate(i, norm, log_scale=log_scale):
         temporal_value = np.array(sub[Temporal1].values)[0]
 
         g1 = np.digitize(sub[Spatio1], lng_gird, right=True)
-        g1 = np.where(g1 >= np.max(lng_gird), np.max(lng_gird) - 1, g1)
+        g1 = np.where(g1 >= np.max(lng_gird), np.max(lng_gird) - 1, g1).astype("int")
 
         g2 = np.digitize(sub[Spatio2], lat_gird, right=True)
-        g2 = np.where(g2 >= np.max(lat_gird), np.max(lat_gird) - 1, g2)
+        g2 = np.where(g2 >= np.max(lat_gird), np.max(lat_gird) - 1, g2).astype("int")
 
         sub.loc[:, f"{Spatio1}_grid"] = g1
         sub.loc[:, f"{Spatio2}_grid"] = g2