diff --git a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/FlatFileItemReaderAdvancedTests.java b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/FlatFileItemReaderAdvancedTests.java
index b0f33c8cf..16ba26c3d 100644
--- a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/FlatFileItemReaderAdvancedTests.java
+++ b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/file/FlatFileItemReaderAdvancedTests.java
@@ -152,7 +152,7 @@ public class FlatFileItemReaderAdvancedTests extends TestCase {
 		// get restart data
 		reader.update(executionContext);
 		assertEquals(4, executionContext.getLong(ClassUtils.getShortName(FlatFileItemReader.class)
-		        + ".item.count"));
+		        + ".piece.count"));
 		// close input
 		reader.close(executionContext);
 
@@ -167,7 +167,7 @@ public class FlatFileItemReaderAdvancedTests extends TestCase {
 
 		reader.update(executionContext);
 		assertEquals(6, executionContext.getLong(ClassUtils.getShortName(FlatFileItemReader.class)
-		        + ".item.count"));
+		        + ".piece.count"));
 	}
 
 }
diff --git a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/xml/StaxEventItemReaderTests.java b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/xml/StaxEventItemReaderTests.java
index dafb9da52..cce5402cb 100644
--- a/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/xml/StaxEventItemReaderTests.java
+++ b/spring-batch-infrastructure/src/test/java/org/springframework/batch/item/xml/StaxEventItemReaderTests.java
@@ -116,7 +116,7 @@ public class StaxEventItemReaderTests extends TestCase {
 		source.read();
 		source.update(executionContext);
 		System.out.println(executionContext);
-		assertEquals(1, executionContext.getLong(ClassUtils.getShortName(StaxEventItemReader.class) + ".item.count"));
+		assertEquals(1, executionContext.getLong(ClassUtils.getShortName(StaxEventItemReader.class) + ".piece.count"));
 		List expectedAfterRestart = (List) source.read();
 
 		source = createNewInputSouce();
@@ -183,7 +183,7 @@ public class StaxEventItemReaderTests extends TestCase {
 	}
 
 	private long extractRecordCount() {
-		return executionContext.getLong(ClassUtils.getShortName(StaxEventItemReader.class) + ".item.count");
+		return executionContext.getLong(ClassUtils.getShortName(StaxEventItemReader.class) + ".piece.count");
 	}
 
 	public void testCloseWithoutOpen() throws Exception {