fix highlighting on block removal