Render edits properly

I commented on a post via Element, and then edited the post in Element, but it didn't quite render correctly on the demo page:

Looks like it handled the edit logic on the first message but still shhow