fix

pyth-network · Nov 16, 2023 · ef25a26 · ef25a26
1 parent a1db6a4
commit ef25a26
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/constants/flags.ts b/constants/flags.ts
@@ -13,5 +13,6 @@ export const DELEGATOR_BATCH_VOTE_SUPPORT_BY_PLUGIN: Record<
   HeliumVSR: false,
   gateway: false,
   NFT: false,
+  pyth: false,
   unknown: false,
 }
diff --git a/utils/uiTypes/VotePlugin.ts b/utils/uiTypes/VotePlugin.ts
@@ -165,10 +165,16 @@ export class VotingClient {
   ): Promise<ProgramAddresses | undefined> => {
     if (this.noClient) return
     const realm = this.realm!
-    const torAccount = await fetchTokenOwnerRecordByPubkey(
-      this.client!.stakeConnection.program.provider.connection, // FIX ME
-      tokenOwnerRecord
-    )
+    const torAccount =
+      this.client instanceof PythClient
+        ? await fetchTokenOwnerRecordByPubkey(
+            this.client.stakeConnection.program.provider.connection,
+            tokenOwnerRecord
+          )
+        : await fetchTokenOwnerRecordByPubkey(
+            this.client!.program.provider.connection,
+            tokenOwnerRecord
+          )
     console.log(this.client)
 
     if (!torAccount.result) return