diff --git a/Assignment-2/Assignment-2.h b/Assignment-2/Assignment-2.h
index 38e9787..a00dd30 100644
--- a/Assignment-2/Assignment-2.h
+++ b/Assignment-2/Assignment-2.h
@@ -34,7 +34,7 @@ using namespace SVF;
 class ICFGTraversal
 {
 public:
-    typedef std::vector<const SVFInstruction*> CallStack;
+    typedef std::vector<const ICFGNode*> CallStack;
 
 public: