1 files changed, 3 insertions, 3 deletions
diff --git a/include/cps/AST.h b/include/cps/AST.h
index 5aadc21..c7764c6 100644
--- a/include/cps/AST.h
+++ b/include/cps/AST.h
@@ -170,10 +170,10 @@ public:
 };
 
 class InputStmtAST : public StmtAST {
-    std::string Name;
+    std::unique_ptr<ExprAST> Target;
 public:
-    InputStmtAST(const std::string &Name) : Name(Name) {}
-    const std::string &getName() const { return Name; }
+    InputStmtAST(std::unique_ptr<ExprAST> Target) : Target(std::move(Target)) {}
+    ExprAST *getTarget() const { return Target.get(); }
 };
 
 class OutputStmtAST : public StmtAST {