diff --git a/diagram.py b/diagram.py
index 514a2ad..08c7cab 100644
--- a/diagram.py
+++ b/diagram.py
@@ -60,16 +60,16 @@ class GraphDef(metaclass=PoolMeta):
             return None
 
         if self.scaling == 'alldata':
-            query = [('asset.id', '=', self.asset.id)]
+            query = [('asset', '=', self.asset.id)]
         elif self.scaling == 'view':
             query = [
-                ('asset.id', '=', self.asset.id),
+                ('asset', '=', self.asset.id),
                 ('date', '>=', self.chart.used_start_date()),
                 ('date', '<=', self.chart.used_end_date()),
                 ]
         elif self.scaling == 'six':
             query = [
-                ('asset.id', '=', self.asset.id),
+                ('asset', '=', self.asset.id),
                 ('date', '>=', self.chart.used_start_date() -
                     timedelta(days=180)),
                 ('date', '<=', self.chart.used_end_date()),
@@ -104,12 +104,12 @@ class ChartPoint(metaclass=PoolMeta):
 
             before = Rate.search([
                 ('date', '<', query_date),
-                ('asset.id', '=', asset_id),
+                ('asset', '=', asset_id),
                 ], limit=1, order=[('date', 'DESC')])
 
             after = Rate.search([
                 ('date', '>', query_date),
-                ('asset.id', '=', asset_id),
+                ('asset', '=', asset_id),
                 ], limit=1, order=[('date', 'ASC')])
 
             if (len(before) == 1) and (len(after) == 1):