--- a/pallets/refungible/src/lib.rs
+++ b/pallets/refungible/src/lib.rs
@@ -712,7 +712,6 @@
 	}
 
 	fn total_pieces(collection_id: CollectionId, token_id: TokenId) -> u128 {
-		// Not "try_fold" because total count of pieces is limited by 'MAX_REFUNGIBLE_PIECES'.
-		<Balance<T>>::iter_prefix((collection_id, token_id)).fold(0, |total, piece| total + piece.1)
+		<TotalSupply<T>>::get((collection_id, token_id))
 	}
 }
--- a/tests/src/refungible.test.ts
+++ b/tests/src/refungible.test.ts
@@ -68,6 +68,27 @@
     });
   });
 
+  it.only('Check total pieces of token', async () => {
+    await usingApi(async api => {
+      const createCollectionResult = await createCollection(api, alice, {mode: {type: 'ReFungible'}});
+      expect(createCollectionResult.success).to.be.true;    
+      const collectionId  = createCollectionResult.collectionId;
+      const amountPieces = 100n;
+      const result = await createRefungibleToken(api, alice, collectionId, amountPieces);
+      expect(result.success).to.be.true;
+      {
+        const totalPieces = await api.rpc.unique.totalPieces(collectionId, result.itemId);
+        expect(totalPieces.toBigInt()).to.be.eq(amountPieces);
+      }
+
+      await transfer(api, collectionId, result.itemId, alice, bob, 60n);
+      {
+        const totalPieces = await api.rpc.unique.totalPieces(collectionId, result.itemId);
+        expect(totalPieces.toBigInt()).to.be.eq(amountPieces);
+      }
+    });
+  });
+
   it('Transfer token pieces', async () => {
     await usingApi(async api => {
       const collectionId = (await createCollection(api, alice, {mode: {type: 'ReFungible'}})).collectionId;