后来武则天想立自己的侄子武三思为太子，狄仁杰知道后冒死相谏"今天下仍念唐朝恩德，若立太子，非庐陵王李显不可"。武则天听后很不高兴。狄仁杰继续说道"先帝将二子托于陛下，陛下现在却要把天下让给外姓吗？而且武三思只是陛下的侄子，若他为帝，您凭何入太庙？愿陛下慎重啊。"
武则天逐渐醒悟，于是把李显召回京城。到了公元700年，狄仁杰病重，武则天向来尊重他，称他为"国老"，于是亲自来看望他，命宫廷最好的御医医治。只可惜狄仁杰后来还是去世了，临死前他含泪对武则天说了九个字"恢复李唐，臣虽死无憾"，武则天默然点了点头。
