diff --git a/src/test/java/org/ballerinalang/distribution/UpdateToolTest.java b/src/test/java/org/ballerinalang/distribution/UpdateToolTest.java
index e3fcc03..cd251f8 100644
--- a/src/test/java/org/ballerinalang/distribution/UpdateToolTest.java
+++ b/src/test/java/org/ballerinalang/distribution/UpdateToolTest.java
@@ -287,6 +287,7 @@ public void testRemoveCommand() throws IOException, InterruptedException {
         output = TestUtils.executeCommand(args);
         Assert.assertTrue(output.contains("successfully removed"));
         Assert.assertFalse(Files.exists(TestUtils.getDistPath(previousChannelVersion)));
+        Assert.assertNull(TestUtils.getDependencyPath(previousChannelDependencyVersion));
 
         args.add("arg1");
         output = TestUtils.executeCommand(args);
@@ -301,7 +302,7 @@ public void testRemoveCommand() throws IOException, InterruptedException {
         Assert.assertTrue(Files.exists(TestUtils.getDistPath(swanLakeLatestVersion)));
         Assert.assertFalse(Files.exists(TestUtils.getDistPath(swanLakeVersion)));
         Assert.assertFalse(Files.exists(TestUtils.getDistPath(previousChanneLatestVersion)));
-        Assert.assertTrue(Files.exists(TestUtils.getDependencyPath(swanLakeLatestVersionDependency)));
+        Assert.assertNull(TestUtils.getDependencyPath(swanLakeLatestVersionDependency));
         Assert.assertEquals(Files.list(TestUtils.getDependencyPath(swanLakeLatestVersionDependency).getParent()).count()
                 , 1);