diff --git a/tests/CMSMainTest.php b/tests/CMSMainTest.php
index df89c3c6..4bd337d7 100644
--- a/tests/CMSMainTest.php
+++ b/tests/CMSMainTest.php
@@ -128,7 +128,7 @@ class CMSMainTest extends FunctionalTest {
$this->assertTrue($livePage->canDelete());
// Check that the 'restore' button exists as a simple way of checking that the correct page is returned.
- $this->assertRegExp('/]+type="submit"[^>]+name="action_restore"/i', $response->getBody());
+ $this->assertRegExp('/]+type="submit"[^>]+name="action_(restore|revert)"/i', $response->getBody());
}
/**