Fixing comments edit