diff --git a/src/SaveHandler/DbTableGateway.php b/src/SaveHandler/DbTableGateway.php
index 9865c756..659148cd 100644
--- a/src/SaveHandler/DbTableGateway.php
+++ b/src/SaveHandler/DbTableGateway.php
@@ -129,9 +129,9 @@ public function write($id, $data)
         $rows = $this->tableGateway->select([
             $this->options->getIdColumn()   => $id,
             $this->options->getNameColumn() => $this->sessionName,
-        ]);
+        ])->current();
 
-        if ($rows->current()) {
+        if ($rows) {
             return (bool) $this->tableGateway->update($data, [
                 $this->options->getIdColumn()   => $id,
                 $this->options->getNameColumn() => $this->sessionName,