After replace all, cells un-highlighted

Not sure of desired behavior. This changes is easily reverted by
removing the two lines of code indicated with comment.