diff --git a/contracts/senders/aave-v2-migrator/main.sol b/contracts/senders/aave-v2-migrator/main.sol index 13201cc..659f1cc 100644 --- a/contracts/senders/aave-v2-migrator/main.sol +++ b/contracts/senders/aave-v2-migrator/main.sol @@ -164,7 +164,8 @@ contract MigrateResolver is LiquidityResolver { } function migrateWithFlash(AaveDataRaw calldata _data, uint ethAmt) external { - bytes memory data = abi.encode(_data, msg.sender, ethAmt); + bytes memory callbackData = abi.encodeWithSelector(bytes4(this.migrateFlashCallback.selector), _data, msg.sender, ethAmt); + bytes memory data = abi.encode(callbackData, ethAmt); flashloanContract.initiateFlashLoan(data, ethAmt); }