diff --git a/src/pages/PersonalizationAI/PersonalizationAI.tsx b/src/pages/PersonalizationAI/PersonalizationAI.tsx
index 84ca429f..1476a6b3 100644
--- a/src/pages/PersonalizationAI/PersonalizationAI.tsx
+++ b/src/pages/PersonalizationAI/PersonalizationAI.tsx
@@ -242,7 +242,7 @@ export default function PersonalizationAI() {
       if (payError?.includes("insufficient funds") || payError?.includes("Payment Required")) {
         var link = document.createElement("a");
         link.href = generateHubWalletRequestURL({
-          wayback: "edit",
+          wayback: "personalization-ai",
           action: "buy",
           dif: "50000",
           userid: userId,
diff --git a/src/utils/hooks/useAutoPay.ts b/src/utils/hooks/useAutoPay.ts
index 7c1d9956..116fb66e 100644
--- a/src/utils/hooks/useAutoPay.ts
+++ b/src/utils/hooks/useAutoPay.ts
@@ -28,6 +28,14 @@ export const useAfterPay = () => {
   let URLadditionalinformation = searchParams.get("additionalinformation");//его токен
 
   useEffect(() => {
+    console.log("useAutoPay: Processing return from payment", {
+      URLaction,
+      URLuserId,
+      URLadditionalinformation,
+      userId,
+      wayback: searchParams.get("wayback")
+    });
+    
     setSearchParams({}, { replace: true });
 
     if (userId && URLuserId && userId === URLuserId) {
@@ -53,12 +61,32 @@ export const useAfterPay = () => {
 
 
       //Покупка ИИ тарифа из настройки квиза ИИ вкладки
-      if (location.pathname === "/personalization-ai") {
+      if (URLaction === "buy" && URLadditionalinformation) {
         const quizId = Number(URLadditionalinformation);
+        console.log("useAutoPay: Processing AI tariff purchase", {
+          quizId,
+          wayback: searchParams.get("wayback")
+        });
+        
         if (quizId) {
           setEditQuizId(Number(quizId)); //Выбираем квиз
           setCurrentStep(17); // Шаг для персонализации AI
-          navigate("/personalization-ai"); //Непосредственно переход
+          
+          // Проверяем wayback параметр для определения куда переходить
+          const wayback = searchParams.get("wayback");
+          if (wayback === "edit") {
+            console.log("useAutoPay: Navigating to /edit first, then /personalization-ai");
+            // Сначала переходим на /edit, затем на /personalization-ai
+            navigate("/edit");
+            // Используем setTimeout чтобы дать время для загрузки /edit
+            setTimeout(() => {
+              navigate("/personalization-ai");
+            }, 100);
+          } else {
+            console.log("useAutoPay: Direct navigation to /personalization-ai");
+            // Прямой переход на /personalization-ai
+            navigate("/personalization-ai");
+          }
         }
       }