diff --git a/image_chromeos.py b/image_chromeos.py
index 1e061cd..17588d9 100755
--- a/image_chromeos.py
+++ b/image_chromeos.py
@@ -133,16 +133,31 @@
     chroot_image = os.path.join(
         "..",
         located_image[len(real_src_dir):].lstrip("/"))
-    cros_image_to_target_args = ["--remote=%s" % options.remote,
-                                 "--board=%s" % board,
-                                 "--from=%s" % os.path.dirname(chroot_image),
-                                 "--image-name=%s" %
-                                 os.path.basename(located_image)]
 
-    command = ("./bin/cros_image_to_target.py %s" %
-               " ".join(cros_image_to_target_args))
-    if options.image_args:
-      command += " %s" % options.image_args
+    # Check to see if cros flash is in the chroot or not.
+    cros_flash_path = os.path.join(options.chromeos_root,
+                                   "chromite/cros/commands/cros_flash.py")
+    if os.path.exists(cros_flash_path):
+      # Use 'cros flash'
+      cros_flash_args = ["--board=%s" % board,
+                         "--clobber-stateful",
+                         options.remote,
+                         chroot_image]
+
+      command = ("cros flash %s" % " ".join(cros_flash_args))
+    else:
+      # Use 'cros_image_to_target.py'
+
+      cros_image_to_target_args = ["--remote=%s" % options.remote,
+                                   "--board=%s" % board,
+                                   "--from=%s" % os.path.dirname(chroot_image),
+                                   "--image-name=%s" %
+                                   os.path.basename(located_image)]
+
+      command = ("./bin/cros_image_to_target.py %s" %
+                 " ".join(cros_image_to_target_args))
+      if options.image_args:
+        command += " %s" % options.image_args
 
     # Workaround for crosbug.com/35684.
     os.chmod(misc.GetChromeOSKeyFile(options.chromeos_root), 0600)