diff --git a/tests/controller/CMSMainTest.php b/tests/controller/CMSMainTest.php
index 5af13188..1409de33 100644
--- a/tests/controller/CMSMainTest.php
+++ b/tests/controller/CMSMainTest.php
@@ -247,7 +247,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('/<button[^>]+name="action_(restore|revert)"/i', $response->getBody());
+		$this->assertRegExp('/<input type="submit"[^>]+name="action_(restore|revert)"/i', $response->getBody());
 	}
 
 	/**