diff --git a/migrations/7_pol_migration.js b/migrations/7_pol_migration.js
index 05fa06f6..87f9cbd3 100644
--- a/migrations/7_pol_migration.js
+++ b/migrations/7_pol_migration.js
@@ -1,4 +1,5 @@
 const ethUtils = require('ethereumjs-util')
+const assert = require('assert')
 
 const utils = require('./utils')
 const contractAddresses = require('../contractAddresses.json')
@@ -73,6 +74,9 @@ async function migrateMatic(governance, depositManager, mintAmount) {
     depositManager.contract.methods.migrateMatic(mintAmount).encodeABI()
   )
   console.log('Migrate MATIC tokens to POL tokens:', result.tx)
+
+  const newDepositManagerPOLBalance = await polToken.contract.methods.balanceOf(newDepositManager.address).call()
+  assertBigNumberEquality(newDepositManagerPOLBalance, maticAmountToMintAndMigrateInDepositManager)
   */
 }
 
@@ -104,9 +108,6 @@ module.exports = async function(deployer, _, _) {
     const maticAmountToMintAndMigrateInDepositManager = oneEther.mul(web3.utils.toBN('1000000000')).toString() // 100 ethers
     await migrateMatic(governance, newDepositManager, maticAmountToMintAndMigrateInDepositManager)
 
-    const newDepositManagerPOLBalance = await polToken.contract.methods.balanceOf(newDepositManager.address).call()
-    assertBigNumberEquality(newDepositManagerPOLBalance, maticAmountToMintAndMigrateInDepositManager)
-
     // Update contract addresses.
     contractAddresses.root.NewDepositManager = newDepositManager.address
     contractAddresses.root.PolToken = polToken.address