fixed some remove display issue