diff --git a/tests/test_hub/test_get_last_host_update.py b/tests/test_hub/test_get_last_host_update.py
index b304f5da..4ace4c4f 100644
--- a/tests/test_hub/test_get_last_host_update.py
+++ b/tests/test_hub/test_get_last_host_update.py
@@ -28,7 +28,7 @@ class TestGetLastHostUpdate(DBQueryTestCase):
         self.assertEqual(query.joins, ['host ON sessions.user_id = host.user_id'])
         self.assertEqual(query.clauses, ['host.id = %(hostID)i'])
         self.assertEqual(query.values, {'hostID': 1})
-        self.assertEqual(query.columns, ['update_time'])
+        self.assertEqual(query.columns, ['sessions.update_time'])
 
     def test_valid_datetime(self):
         if sys.version_info[1] <= 6:
diff --git a/tests/test_hub/test_get_ready_hosts.py b/tests/test_hub/test_get_ready_hosts.py
index 8c1447ae..7ae8843f 100644
--- a/tests/test_hub/test_get_ready_hosts.py
+++ b/tests/test_hub/test_get_ready_hosts.py
@@ -45,7 +45,7 @@ class TestGetReadyHosts(unittest.TestCase):
                                        'host_config ON host.id = host_config.host_id'])
         self.assertEqual(query.clauses, ['active IS TRUE', 'enabled IS TRUE', 'expired IS FALSE',
                                          'master IS NULL', 'ready IS TRUE',
-                                         "update_time > NOW() - '5 minutes'::interval"])
+                                         "sessions.update_time > NOW() - '5 minutes'::interval"])
         self.assertEqual(query.values, {})
         self.assertEqual(query.columns, ['arches', 'capacity', 'host.id', 'name', 'task_load'])