From f3a9f869d18298f8bf3f2ec42c08c9a82bbfd5eb Mon Sep 17 00:00:00 2001
From: Ryan Levy <rlevy@flatironinstitute.org>
Date: Wed, 18 Dec 2024 08:40:02 -0500
Subject: [PATCH] Fix initial pivot

---
 ext/ITensorNumericalAnalysisTCIExt/tci_util.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ext/ITensorNumericalAnalysisTCIExt/tci_util.jl b/ext/ITensorNumericalAnalysisTCIExt/tci_util.jl
index 59459d5..fd4ff49 100644
--- a/ext/ITensorNumericalAnalysisTCIExt/tci_util.jl
+++ b/ext/ITensorNumericalAnalysisTCIExt/tci_util.jl
@@ -38,7 +38,8 @@ function ITensorTCI.interpolate(
     initial_pivot = random_initial_pivot(s_renamed)
   else
     # manually rename
-    initial_pivot = [forward_dict[v] => initial_pivot[v] for v in vertices(s)]
+    # assuming from calculate_ind_values
+    initial_pivot = [forward_dict[v] => initial_pivot[only(s[v])] + 1 for v in vertices(s)]
   end
 
   tn = ITensorTCI.interpolate(