fix rewards only preferential gauges

balancer · Sep 5, 2023 · ea82261 · ea82261
1 parent 91892e4
commit ea82261
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gaugeFactory.ts b/src/gaugeFactory.ts
@@ -109,16 +109,18 @@ export function handleRewardsOnlyGaugeCreated(
   gauge.pool = poolRegistered ? poolAddress.toHexString() : null;
   gauge.poolId = poolRegistered ? getPoolId(poolAddress) : null;
   gauge.factory = factory.id;
-  gauge.save();
 
   if (poolRegistered) {
     let pool = getPoolEntity(poolAddress, event.params.gauge);
     pool.address = poolAddress;
     pool.poolId = getPoolId(poolAddress);
     pool.preferentialGauge = gauge.id;
+    gauge.isPreferentialGauge = true;
     pool.save();
   }
 
+  gauge.save();
+
   RewardsOnlyGaugeTemplate.create(event.params.gauge);
   ChildChainStreamerTemplate.create(event.params.streamer);
 }