Edit Contribution form for Membership shows wrong line item

The Edit form of contribution related to membership shows in correct line item. But when we try to save the contribution the line items (financial item) are updated correctly. The code to fetch line items for participant and membership for contribution needs to be fetched by using contribution id rather then contribution id. I guess the fix have had previously been fixed for contribution view form.

To replicate:
1. Create a price set for membership with and without membership type of type checkbox.
2. Create Back-end membership for a contact and select all option and record payment.
3. Edit the contribution associated with the membership.