Fixing the comments to refresh when updated