diff --git a/code_diff/ast.py b/code_diff/ast.py
index 53f18bb..9378e0a 100644
--- a/code_diff/ast.py
+++ b/code_diff/ast.py
@@ -1,4 +1,4 @@
-import code_ast as ct
+import code_tokenize as ct
 
 from collections import defaultdict