diff --git a/tests/CMSMainTest.php b/tests/CMSMainTest.php
index 48eac79d..b7c237d4 100644
--- a/tests/CMSMainTest.php
+++ b/tests/CMSMainTest.php
@@ -47,8 +47,8 @@ class CMSMainTest extends FunctionalTest {
 			$response = Director::test("admin/cms/batchactions/publish", array('csvIDs' => implode(',', array($page1->ID, $page2->ID)), 'ajax' => 1), $this->session());
 	
 			$responseData = Convert::json2array($response->getBody());
-			$this->assertTrue(property_exists($responseData['modified'], '1'));
-			$this->assertTrue(property_exists($responseData['modified'], '2'));
+			$this->assertTrue(property_exists($responseData['modified'], $page1->ID));
+			$this->assertTrue(property_exists($responseData['modified'], $page2->ID));
 		}
 
 		// Get the latest version of the redirector page