diff --git a/src/main/java/im/rosetta/database/repository/PushTokenRepository.java b/src/main/java/im/rosetta/database/repository/PushTokenRepository.java
index eb1e990..854ac40 100644
--- a/src/main/java/im/rosetta/database/repository/PushTokenRepository.java
+++ b/src/main/java/im/rosetta/database/repository/PushTokenRepository.java
@@ -50,7 +50,7 @@ public class PushTokenRepository extends Repository<PushToken> {
     public PushToken findToken(Device device, String token){
         try (Session session = HibernateUtil.getSessionFactory().openSession()) {
             return session.createQuery("SELECT pt FROM PushToken pt WHERE pt.device = :device AND pt.token = :token", PushToken.class)
-                    .setParameter("device", device.getDeviceId())
+                    .setParameter("device", device)
                     .setParameter("token", token)
                     .uniqueResult();
         }