diff --git a/scripts/generate_equation_implication_js.py b/scripts/generate_equation_implication_js.py
index baf7e670..02565f13 100644
--- a/scripts/generate_equation_implication_js.py
+++ b/scripts/generate_equation_implication_js.py
@@ -110,7 +110,7 @@ def find_equivalence_classes_fast(implications):
 
 special = []
 for line in open("equational_theories/Equations.lean"):
-    if line.startswith("abbrev Equation"):
+    if line.startswith("equation") and ':=' in line:
         special.append(line.split()[1])
 print("var special = ", special)