diff --git a/spark/sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java b/spark/sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java
index 59d5dcf0b..02bfb47e6 100644
--- a/spark/sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java
+++ b/spark/sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java
@@ -95,7 +95,8 @@ public void testBasicRead() throws Exception {
 		assertEquals(5, dataFrame.take(5).length);
 		DataFrame results = sqc
 				.sql("SELECT name FROM datfile WHERE id >=1 AND id <=10");
-		assertEquals(10, dataFrame.take(10).length);
+		assertEquals(10, results.take(10).length);
+		assertEquals(10L, results.count());
 	}
 
 	@Test