diff --git a/Assignment-3/Test3.cpp b/Assignment-3/Test3.cpp
index 0aa9f50..171f475 100644
--- a/Assignment-3/Test3.cpp
+++ b/Assignment-3/Test3.cpp
@@ -49,6 +49,7 @@ void Test1()
     delete andersenPTA; 
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
     SVF::SVFIR::releaseSVFIR();
+    NodeIDAllocator::unset();
 }
 
 void Test2()
@@ -65,6 +66,7 @@ void Test2()
     delete andersenPTA; 
     SVF::SVFIR::releaseSVFIR();
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
+    NodeIDAllocator::unset();
 }
 
 void Test3()
@@ -80,6 +82,7 @@ void Test3()
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
     SVF::SVFIR::releaseSVFIR();
     delete andersenPTA; 
+    NodeIDAllocator::unset();
 }
 void Test()
 {
diff --git a/Assignment-4/Test4.cpp b/Assignment-4/Test4.cpp
index 5e6c460..197cd9d 100644
--- a/Assignment-4/Test4.cpp
+++ b/Assignment-4/Test4.cpp
@@ -50,6 +50,7 @@ void Test1()
     cout << "\n test1 passed !" << endl;
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
     SVF::SVFIR::releaseSVFIR();
+    NodeIDAllocator::unset();
 }
 void Test2()
 {
@@ -66,6 +67,7 @@ void Test2()
     cout << "\n test2 passed !" << endl;
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
     SVF::SVFIR::releaseSVFIR();
+    NodeIDAllocator::unset();
 }
 
 void Test3()
@@ -83,7 +85,7 @@ void Test3()
     cout << "\n test3 passed !" << endl;
     SVF::SVFIR::releaseSVFIR();
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
-
+    NodeIDAllocator::unset();
 }
 
 void Test4()
@@ -102,6 +104,7 @@ void Test4()
     cout << "\n test4 passed !" << endl;
     SVF::LLVMModuleSet::releaseLLVMModuleSet();
     SVF::SVFIR::releaseSVFIR();
+    NodeIDAllocator::unset();
 }
 int main(int argc, char ** argv)
 {