diff --git a/src/main/java/im/rosetta/database/repository/PushTokenRepository.java b/src/main/java/im/rosetta/database/repository/PushTokenRepository.java
index 97aea0f..eb1e990 100644
--- a/src/main/java/im/rosetta/database/repository/PushTokenRepository.java
+++ b/src/main/java/im/rosetta/database/repository/PushTokenRepository.java
@@ -49,7 +49,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_id = :device AND pt.token = :token", PushToken.class)
+            return session.createQuery("SELECT pt FROM PushToken pt WHERE pt.device = :device AND pt.token = :token", PushToken.class)
                     .setParameter("device", device.getDeviceId())
                     .setParameter("token", token)
                     .uniqueResult();