diff --git a/test/Adapter/DbTableGatewayTest.php b/test/Adapter/DbTableGatewayTest.php
index d2607ad..98e5236 100644
--- a/test/Adapter/DbTableGatewayTest.php
+++ b/test/Adapter/DbTableGatewayTest.php
@@ -10,6 +10,7 @@
 namespace ZendTest\Paginator\Adapter;
 
 use Zend\Db\Adapter\Platform\Sql92;
+use Zend\Paginator\Adapter\DbSelect;
 use Zend\Paginator\Adapter\DbTableGateway;
 use Zend\Db\ResultSet\ResultSet;
 
@@ -74,7 +75,7 @@ public function testCount()
         $mockResult = $this->getMock('Zend\Db\Adapter\Driver\ResultInterface');
         $mockResult->expects($this->any())
                    ->method('current')
-                   ->will($this->returnValue(array('c' => 10)));
+                   ->will($this->returnValue(array(DbSelect::ROW_COUNT_COLUMN_NAME => 10)));
 
         $this->mockStatement->expects($this->any())
              ->method('execute')