Update x/superfluid/keeper/stake_test.go

Co-authored-by: Matt, Park <45252226+mattverse@users.noreply.github.com>
osmosis-labs · Mar 8, 2022 · d911b90 · d911b90
1 parent a594d4e
commit d911b90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/x/superfluid/keeper/stake_test.go b/x/superfluid/keeper/stake_test.go
@@ -47,8 +47,8 @@ func (suite *KeeperTestSuite) SetupSuperfluidDelegations(delAddrs []sdk.AccAddre
 		delAddr := delAddrs[del.delIndex]
 		valAddr := valAddrs[del.valIndex]
 		lock := suite.SetupSuperfluidDelegate(delAddr, valAddr, denoms[del.lpIndex], del.lpAmount)
-		expAcc := types.NewSuperfluidIntermediaryAccount(lock.Coins[0].Denom, valAddr.String(), 0)
-		gotAcc := suite.App.SuperfluidKeeper.GetIntermediaryAccount(suite.Ctx, expAcc.GetAccAddress())
+		address := suite.App.SuperfluidKeeper.GetLockIdIntermediaryAccountConnection(suite.Ctx, lock.ID)
+		gotAcc := suite.App.SuperfluidKeeper.GetIntermediaryAccount(suite.Ctx, address)
 
 		// save accounts for future use
 		if flagIntermediaryAcc[gotAcc.String()] == false {