From 9b76b421e442f4cbce1a4298e22d3d0b0184dc51 Mon Sep 17 00:00:00 2001
From: Chris Sosa <sosa@chromium.org>
Date: Thu, 28 Oct 2010 03:41:14 -0700
Subject: [PATCH] Use the right base path for vm updates.

Change-Id: I98276b26b85e48ae4b65e8f3fb65540b24665f9a

BUG=Will file
TEST=Ran it

Review URL: http://codereview.chromium.org/4185008
---
 bin/cros_au_test_harness.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/bin/cros_au_test_harness.py b/bin/cros_au_test_harness.py
index 9036dc95b2..d88b353a72 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,