diff --git a/executor/cli/git.py b/executor/cli/git.py
index 183300a..ff13add 100644
--- a/executor/cli/git.py
+++ b/executor/cli/git.py
@@ -8,7 +8,8 @@ import base64
 import json
 
 
-from executor.engine import Engine
+from executor.visitor import Visitor
+from executor.state import State
 from executor.core import config
 from executor.handlers.config import handle_config