diff --git a/lib/livebook/teams/org.ex b/lib/livebook/teams/org.ex
index d5007f632..da77f9886 100644
--- a/lib/livebook/teams/org.ex
+++ b/lib/livebook/teams/org.ex
@@ -37,8 +37,13 @@ defmodule Livebook.Teams.Org do
   @doc """
   Generates a hash key.
   """
-  @spec key_hash(%{teams_key: String.t()}) :: String.t()
+  @spec key_hash(%{teams_key: String.t()} | String.t()) :: String.t()
+
   def key_hash(%{teams_key: teams_key}) do
+    key_hash(teams_key)
+  end
+
+  def key_hash(teams_key) do
     hash = :crypto.hash(:sha256, teams_key)
     Base.url_encode64(hash, padding: false)
   end