Index: bin/cros_au_test_harness.py |
diff --git a/bin/cros_au_test_harness.py b/bin/cros_au_test_harness.py |
index 9036dc95b21ec85148317308b226ba99ecca8fb4..d88b353a72de825bfa1ffc781d00b8bb4b6d0331 100755 |
--- a/bin/cros_au_test_harness.py |
+++ b/bin/cros_au_test_harness.py |
@@ -252,6 +252,8 @@ class VirtualAUTest(unittest.TestCase, AUTest): |
def UpdateImage(self, image_path, stateful_change='old'): |
"""Updates VM image with image_path.""" |
stateful_change_flag = self.GetStatefulChangeFlag(stateful_change) |
+ if self.source_image == base_image_path: |
+ self.source_image = self.vm_image_path |
RunCommand(['%s/cros_run_vm_update' % self.crosutilsbin, |
'--update_image_path=%s' % image_path, |