From 674f27a765ea29f18f788db495c36c43feddd454 Mon Sep 17 00:00:00 2001
From: Artem Kovalchuk <kovart.dev@gmail.com>
Date: Mon, 12 Feb 2024 20:12:19 +0200
Subject: [PATCH] fix rewardsDistributor tests

---
 subgraph/src/tests/rewardsDistributor.test.ts | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/subgraph/src/tests/rewardsDistributor.test.ts b/subgraph/src/tests/rewardsDistributor.test.ts
index 4742b2a6..1a40ac81 100644
--- a/subgraph/src/tests/rewardsDistributor.test.ts
+++ b/subgraph/src/tests/rewardsDistributor.test.ts
@@ -101,9 +101,7 @@ describe("Rewards distributor", () => {
   test("should handle a reward event on a pool with one delegator and return the correct APY", () => {
     //Given
     delegatorOne = new Staker(mockDelegatorStakerIdOne);
-    (delegatorOne as any).account = mockDelegatorStakerIdOne;
-    (delegatorOne as any).aggregateActiveStake = "";
-    (delegatorOne as any).aggregateTotalStake = "";
+    delegatorOne.account = mockDelegatorStakerIdOne;
     delegatorOne.nodePools = [mockPoolId];
     delegatorOne.save();
 
@@ -164,15 +162,11 @@ describe("Rewards distributor", () => {
     //Given
     delegatorOne = new Staker(mockDelegatorStakerIdOne);
     delegatorOne.account = mockDelegatorStakerIdOne;
-    (delegatorOne as any).aggregateActiveStake = "";
-    (delegatorOne as any).aggregateTotalStake = "";
     delegatorOne.nodePools = [mockPoolId];
     delegatorOne.save();
 
     delegatorTwo = new Staker(mockDelegatorStakerIdTwo);
     delegatorTwo.account = mockDelegatorStakerIdTwo;
-    (delegatorTwo as any).aggregateActiveStake = "";
-    (delegatorTwo as any).aggregateTotalStake = "";
     delegatorTwo.nodePools = [mockPoolId];
     delegatorTwo.save();