Restored comments show up in wrong section

First off, thank you for this feature! Loved it back on Relay for reddit.

To reproduce the bug: Start a reply on comment A Back out of it Start a reply on comment B Your previously saved reply on comment A shows up on comment B