狄仁杰在武则天时期身兼数职。所到之处都深受百姓爱戴，因此狄仁杰也多次获得武则天的嘉奖。有一次，七十多岁的武则天做了一个梦，梦中一只鹦鹉折断了翅膀。她问狄仁杰，狄仁杰微微一笑，给了梦境的解释。武则天听后恍然大悟。

武则天晚年，为继承皇位的事情绞尽脑汁。是吴家继承皇位，还是李家继承皇位。狄仁杰在这个问题上有自己的看法，那就是一定要说服武则天把皇位传给李家，而不是传给武则天的侄子。因此，每当武承嗣要求武则天继承同姓时，狄仁杰就极力争辩。

恰巧武则天需要解梦，狄仁杰说：“鹦鹉与武同姓，两翅指陛下的两个儿子。如果陛下立他的儿子为太子，两翼将重现昔日荣光。”狄仁杰这话，其实是在用武则天的梦想，来劝说武则天立他的儿子为太子，传位给李家。

听了狄仁杰的话，武则天心里也觉得有道理。再加上人心都向着李唐，所以武则天才收回了自己的儿子李哲。公元698年，武则天立李哲为太子，更名。李贤。就这样，吴氏彻底断绝了继承皇位的念头，不久吴承嗣就病逝了。

狄仁杰向武则天推荐了一批得力大臣。几年后帮助李显重建唐朝的就是这群大臣。

公元700年，狄仁杰病逝，享年71岁。