diff --git a/semantic-ast/app/Main.hs b/semantic-ast/app/Main.hs
index 5102b3d41..d1fbeb1da 100644
--- a/semantic-ast/app/Main.hs
+++ b/semantic-ast/app/Main.hs
@@ -4,7 +4,7 @@ module Main (main) where
 
 import AST.Unmarshal as Unmarshal
 import qualified Language.Python.AST as AST
-import qualified TreeSitter.Python as Python
+import qualified Language.Python.Grammar as Python
 import Source.Range
 import Source.Span
 import Data.Aeson (toJSON)