Yes, Will do the necessary changes.
But seeing this, I believe, I may have overlooked the same problem while making changes in the other function.
I will check with the other changes too.