diff --git a/src/ZfcUserDoctrineORM/Mapper/User.php b/src/ZfcUserDoctrineORM/Mapper/User.php
index 5def5f3..c63e06f 100644
--- a/src/ZfcUserDoctrineORM/Mapper/User.php
+++ b/src/ZfcUserDoctrineORM/Mapper/User.php
@@ -3,6 +3,7 @@
 namespace ZfcUserDoctrineORM\Mapper;
 
 use Doctrine\ORM\EntityManagerInterface;
+use ZfcUser\Entity\UserInterface;
 use ZfcUser\Mapper\User as ZfcUserMapper;
 use ZfcUserDoctrineORM\Options\ModuleOptions;
 use Zend\Stdlib\Hydrator\HydratorInterface;
@@ -46,12 +47,12 @@ public function findById($id)
         return $er->find($id);
     }
 
-    public function insert($entity, $tableName = null, HydratorInterface $hydrator = null)
+    public function insert(UserInterface $entity)
     {
         return $this->persist($entity);
     }
 
-    public function update($entity, $where = null, $tableName = null, HydratorInterface $hydrator = null)
+    public function update(UserInterface $entity)
     {
         return $this->persist($entity);
     }