diff --git a/update_attempter.h b/update_attempter.h
index d64771b..dfde4ff 100644
--- a/update_attempter.h
+++ b/update_attempter.h
@@ -145,12 +145,10 @@
   // This is the internal entry point for going through a rollback. This will
   // attempt to run the postinstall on the non-active partition and set it as
   // the partition to boot from. If |powerwash| is True, perform a powerwash
-  // as part of rollback. If |install_path| is set, use this value to determine
-  // the partitions to roll back to (used in testing). Returns True on success.
-  bool Rollback(bool powerwash, std::string* install_path);
+  // as part of rollback. Returns True on success.
+  bool Rollback(bool powerwash);
 
-  // This is the internal entry point for checking if a valid rollback
-  // partition exists.
+  // This is the internal entry point for checking if we can rollback.
   bool CanRollback() const;
 
   // This is the internal entry point for getting a rollback partition name,