Edit agent feature fix

[rounak610]

Description

Related Issues

Solution and Design

Test Plan

Type of change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to change)
• Docs update

Checklist

• My pull request is atomic and focuses on a single change.
• I have read the contributing guide and my code conforms to the guidelines.
• I have documented my changes clearly and comprehensively.
• I have added the required tests.

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.

---

Rounak Bhatia seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}