diff --git a/scripts/find_TODOs.lean b/scripts/find_TODOs.lean
index 3b73281..1165c6b 100644
--- a/scripts/find_TODOs.lean
+++ b/scripts/find_TODOs.lean
@@ -5,7 +5,7 @@ Authors: Joseph Tooby-Smith
 -/
 import Lean
 import Batteries.Data.String.Matcher
-import Mathlib.Init.Data.Nat.Notation
+import Mathlib.Init.Data.Nat.Lemmas
 /-!
 
 # TODO finder