according to mridulm's comments to adjust the code