Fix remove display after comment is posted