diff --git a/test/livebook/zta/basic_auth_test.exs b/test/livebook/zta/basic_auth_test.exs
index 33cb38d95..4d7241f2f 100644
--- a/test/livebook/zta/basic_auth_test.exs
+++ b/test/livebook/zta/basic_auth_test.exs
@@ -1,6 +1,7 @@
 defmodule Livebook.ZTA.BasicAuthTest do
   use ExUnit.Case, async: true
-  use Plug.Test
+  import Plug.Test
+  import Plug.Conn
 
   alias Livebook.ZTA.BasicAuth
 
diff --git a/test/livebook/zta/cloudflare_test.exs b/test/livebook/zta/cloudflare_test.exs
index b18be61ae..be5a55371 100644
--- a/test/livebook/zta/cloudflare_test.exs
+++ b/test/livebook/zta/cloudflare_test.exs
@@ -1,6 +1,7 @@
 defmodule Livebook.ZTA.CloudflareTest do
   use ExUnit.Case, async: true
-  use Plug.Test
+  import Plug.Test
+  import Plug.Conn
 
   alias Livebook.ZTA.Cloudflare
 
diff --git a/test/livebook/zta/google_iap_test.exs b/test/livebook/zta/google_iap_test.exs
index b750e60bc..67c591fcb 100644
--- a/test/livebook/zta/google_iap_test.exs
+++ b/test/livebook/zta/google_iap_test.exs
@@ -1,6 +1,7 @@
 defmodule Livebook.ZTA.GoogleIAPTest do
   use ExUnit.Case, async: true
-  use Plug.Test
+  import Plug.Test
+  import Plug.Conn
 
   alias Livebook.ZTA.GoogleIAP
 
diff --git a/test/livebook/zta/tailscale_test.exs b/test/livebook/zta/tailscale_test.exs
index e8f9a060f..8d904f1e4 100644
--- a/test/livebook/zta/tailscale_test.exs
+++ b/test/livebook/zta/tailscale_test.exs
@@ -1,6 +1,7 @@
 defmodule Livebook.ZTA.TailscaleTest do
   use ExUnit.Case, async: true
-  use Plug.Test
+  import Plug.Test
+  import Plug.Conn
   alias Livebook.ZTA.Tailscale
 
   @moduletag unix: true
diff --git a/test/livebook_web/plugs/user_plug_test.exs b/test/livebook_web/plugs/user_plug_test.exs
index dacb05cd4..97885b570 100644
--- a/test/livebook_web/plugs/user_plug_test.exs
+++ b/test/livebook_web/plugs/user_plug_test.exs
@@ -1,6 +1,7 @@
 defmodule LivebookWeb.UserPlugTest do
   use ExUnit.Case, async: true
-  use Plug.Test
+  import Plug.Test
+  import Plug.Conn
 
   defp call(conn) do
     LivebookWeb.UserPlug.call(conn, LivebookWeb.UserPlug.init([]))