diff --git a/tests/army/community/cluster/snapshot.py b/tests/army/community/cluster/snapshot.py
index 40783ad4ce..66f9d9f598 100644
--- a/tests/army/community/cluster/snapshot.py
+++ b/tests/army/community/cluster/snapshot.py
@@ -49,14 +49,16 @@ class TDTestCase(TBase):
         # split vgroups
         self.splitVGroups()
         self.trimDb()
+        self.checkAggCorrect()
 
         # balance vgroups
         self.balanceVGroupLeader()
 
         # replica to 1
         self.alterReplica(1)
+        self.checkAggCorrect()
 
-        vgids = self.getVGroup()
+        vgids = self.getVGroup(self.db)
         selid = random.choice(vgids)
         self.balanceVGroupLeaderOn(selid)