Fix tests

lib/livebook_web/controllers/error_controller.ex

@@ -1,9 +1,7 @@
 defmodule LivebookWeb.ErrorController do
   use LivebookWeb, :controller
 
-  def show(conn, %{"error" => "unauthorized"}) do
-    conn
-    |> put_status(:unauthorized)
-    |> render("401.html", %{status: 401})
+  def unauthorized(conn, _) do
+    conn |> put_status(:unauthorized) |> render("401.html", %{status: 401})
   end
 end

lib/livebook_web/router.ex

@@ -157,7 +157,7 @@ defmodule LivebookWeb.Router do
   scope "/", LivebookWeb do
     pipe_through :browser
 
-    get "/:error", ErrorController, :show
+    get "/unauthorized", ErrorController, :unauthorized
   end
 
   scope "/" do

test/support/hub_helpers.ex

@@ -300,7 +300,7 @@ defmodule Livebook.HubHelpers do
   end
 
   defp hub_pid(hub) do
-    if pid = GenServer.whereis({:via, Registry, {Livebook.HubsRegistry, hub.id}}) do
+    if pid = Livebook.Hubs.TeamClient.get_pid(hub.id) do
       {:ok, pid}
     end
   end