Can sensitivity persist even after cavity treatment?

I was told that the cold sensation might persist for a while even after the treatment is finished.
I'm not sure if I should stop worrying about it since it will naturally get better over time, or if I need to get treatment again.