diff --git a/packages/schematics/update/migrate/index_spec.ts b/packages/schematics/update/migrate/index_spec.ts
index 8ebf7b9bf2..65516d8038 100644
--- a/packages/schematics/update/migrate/index_spec.ts
+++ b/packages/schematics/update/migrate/index_spec.ts
@@ -6,7 +6,7 @@
  * found in the LICENSE file at https://angular.io/license
  */
 import { virtualFs } from '@angular-devkit/core';
-import { HostTree, VirtualTree } from '@angular-devkit/schematics';
+import { HostTree } from '@angular-devkit/schematics';
 import { SchematicTestRunner, UnitTestTree } from '@angular-devkit/schematics/testing';
 import { map } from 'rxjs/operators';
 
@@ -16,7 +16,7 @@ describe('@schematics/update:migrate', () => {
     '@schematics/update', __dirname + '/../collection.json',
   );
   let host: virtualFs.test.TestHost;
-  let appTree: UnitTestTree = new UnitTestTree(new VirtualTree());
+  let appTree: UnitTestTree = new UnitTestTree(new HostTree());
 
   beforeEach(() => {
     host = new virtualFs.test.TestHost({});
diff --git a/packages/schematics/update/update/index_spec.ts b/packages/schematics/update/update/index_spec.ts
index f72593143a..9315fe1f57 100644
--- a/packages/schematics/update/update/index_spec.ts
+++ b/packages/schematics/update/update/index_spec.ts
@@ -6,7 +6,7 @@
  * found in the LICENSE file at https://angular.io/license
  */
 import { normalize, virtualFs } from '@angular-devkit/core';
-import { HostTree, VirtualTree } from '@angular-devkit/schematics';
+import { HostTree } from '@angular-devkit/schematics';
 import { SchematicTestRunner, UnitTestTree } from '@angular-devkit/schematics/testing';
 import { map } from 'rxjs/operators';
 import * as semver from 'semver';
@@ -36,7 +36,7 @@ describe('@schematics/update', () => {
     '@schematics/update', __dirname + '/../collection.json',
   );
   let host: virtualFs.test.TestHost;
-  let appTree: UnitTestTree = new UnitTestTree(new VirtualTree());
+  let appTree: UnitTestTree = new UnitTestTree(new HostTree());
 
   beforeEach(() => {
     host = new virtualFs.test.TestHost({