diff --git a/test/nerves_hub_web/channels/websocket_test.exs b/test/nerves_hub_web/channels/websocket_test.exs
index bb44c67fd..c62e77449 100644
--- a/test/nerves_hub_web/channels/websocket_test.exs
+++ b/test/nerves_hub_web/channels/websocket_test.exs
@@ -574,7 +574,7 @@ defmodule NervesHubWeb.WebsocketTest do
       # that it is ready to receive updates, which can take a second or two.
       # using this different status allows us to tell the orchestrator to only
       # schedule update for devices that have "finished" connecting
-      eventually assert(device_connection.status == :connected), 3_000
+      eventually assert(Connections.get_latest_for_device(device.id).status == :connected), 3_000
 
       assert recent_datetime(device_connection.established_at)
       assert recent_datetime(device_connection.last_seen_at)