diff --git a/Assignment-2/Test2.cpp b/Assignment-2/Test2.cpp
index b5e31a1..acf5117 100644
--- a/Assignment-2/Test2.cpp
+++ b/Assignment-2/Test2.cpp
@@ -43,7 +43,7 @@ int test1()
     SVFIRBuilder builder(svfModule);
     SVFIR *svfir = builder.build();
 
-    CallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
+    PTACallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
     builder.updateCallGraph(callgraph);
 
     /// ICFG
@@ -75,7 +75,7 @@ int test2()
     SVFIRBuilder builder(svfModule);
     SVFIR *svfir = builder.build();
 
-    CallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
+    PTACallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
     builder.updateCallGraph(callgraph);
 
     /// ICFG
@@ -106,7 +106,7 @@ int test3()
     SVFIRBuilder builder(svfModule);
     SVFIR *svfir = builder.build();
 
-    CallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
+    PTACallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
     builder.updateCallGraph(callgraph);
 
     /// ICFG
diff --git a/Assignment-4/Test4.cpp b/Assignment-4/Test4.cpp
index 9cff17e..48d918c 100644
--- a/Assignment-4/Test4.cpp
+++ b/Assignment-4/Test4.cpp
@@ -69,7 +69,7 @@ int main(int argc, char **argv)
     SVFIRBuilder builder(svfModule);
     SVFIR *svfir = builder.build();
 
-    CallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
+    PTACallGraph* callgraph = AndersenWaveDiff::createAndersenWaveDiff(svfir)->getCallGraph();
     builder.updateCallGraph(callgraph);
 
     /// ICFG