Image Preview in Edit Post

[leefish]

Hi
I am using a template modification posted by RateU to show preview images of uploaded attachments. Link to modification

I received a report from a user that the preview function when editing a post only shows the new attachment image once the post has been updated (i.e. posted on the forum).

If they click preview post before that then the old image is still showing in the preview even though they have chosen a replacement image. It would make more sense if it showed the updated image, though I do realise they have not yet actually attached the replacement - only indicated they wish to do so. I was wondering if this would be something that I could change in the template or if it was part of the Xthreads code?