diff --git a/test/towerops_web/live/agent_live_test.exs b/test/towerops_web/live/agent_live_test.exs
index cc1e45c0..09b06fd7 100644
--- a/test/towerops_web/live/agent_live_test.exs
+++ b/test/towerops_web/live/agent_live_test.exs
@@ -95,7 +95,7 @@ defmodule ToweropsWeb.AgentLiveTest do
       Agents.update_agent_token_heartbeat(agent_token.id, "127.0.0.1")
 
       Towerops.Repo.update_all(
-        Ecto.Query.from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
+        from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
         set: [last_seen_at: three_minutes_ago]
       )
 
@@ -114,7 +114,7 @@ defmodule ToweropsWeb.AgentLiveTest do
       Agents.update_agent_token_heartbeat(agent_token.id, "127.0.0.1")
 
       Towerops.Repo.update_all(
-        Ecto.Query.from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
+        from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
         set: [last_seen_at: ten_minutes_ago]
       )
 
@@ -133,7 +133,7 @@ defmodule ToweropsWeb.AgentLiveTest do
       Agents.update_agent_token_heartbeat(agent_token.id, "127.0.0.1")
 
       Towerops.Repo.update_all(
-        Ecto.Query.from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
+        from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
         set: [last_seen_at: two_hours_ago]
       )
 
@@ -151,7 +151,7 @@ defmodule ToweropsWeb.AgentLiveTest do
       Agents.update_agent_token_heartbeat(agent_token.id, "127.0.0.1")
 
       Towerops.Repo.update_all(
-        Ecto.Query.from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
+        from(t in Agents.AgentToken, where: t.id == ^agent_token.id),
         set: [last_seen_at: two_days_ago]
       )