```\nuint256 balTotalSupply = pool.balancerPool.totalSupply();\nuint256[] memory balances = new uint256[](_vaultTokens.length);\n// Calculate the proportion of the pool balances owned by the polManager\nif (balTotalSupply != 0) {\n    // Calculate the amount of OHM in the pool owned by the polManager\n    // We have to iterate through the tokens array to find the index of OHM\n    uint256 tokenLen = _vaultTokens.length;\n    for (uint256 i; i < tokenLen; ) {\n        uint256 balance = _vaultBalances[i];\n        uint256 polBalance = (balance * balBalance) / balTotalSupply;\n\n\n        balances[i] = polBalance;\n\n\n        unchecked {\n            ++i;\n        }\n    }\n}\n```\n