diff --git a/tests/unit/DatabaseTest.php b/tests/unit/DatabaseTest.php
index e1697b799..2689fa83d 100644
--- a/tests/unit/DatabaseTest.php
+++ b/tests/unit/DatabaseTest.php
@@ -52,7 +52,7 @@ class DatabaseTest extends \PHPUnit_Framework_TestCase
         $payloadExpected ="$point1\n$point2";
 
         $this->mockClient->expects($this->once())
-            ->method('query')
+            ->method('write')
             ->with($this->equalTo($this->db->getName()), $this->equalTo($payloadExpected));
 
         $this->db->writePoints(array($point1, $point2));