diff --git a/lib/livebook/file_system/s3.ex b/lib/livebook/file_system/s3.ex
index 6051cc672..e4755d8ac 100644
--- a/lib/livebook/file_system/s3.ex
+++ b/lib/livebook/file_system/s3.ex
@@ -108,7 +108,7 @@ defmodule Livebook.FileSystem.S3 do
   If the credentials are not specified by the file system, they are
   fetched from environment variables or AWS instance if applicable.
   """
-  @spec credentials(S3.t()) :: S3.credentials()
+  @spec credentials(t()) :: credentials()
   def credentials(%__MODULE__{} = file_system) do
     case {file_system.access_key_id, file_system.secret_access_key} do
       {nil, nil} ->
diff --git a/lib/livebook/teams.ex b/lib/livebook/teams.ex
index 717903dd3..570f72d70 100644
--- a/lib/livebook/teams.ex
+++ b/lib/livebook/teams.ex
@@ -230,7 +230,7 @@ defmodule Livebook.Teams do
   @doc """
   Gets a list of environment variables for a given Hub.
   """
-  @spec get_environment_variables(Team.t()) :: list(Teams.Agent.t())
+  @spec get_environment_variables(Team.t()) :: list(Teams.EnvironmentVariable.t())
   def get_environment_variables(team) do
     TeamClient.get_environment_variables(team.id)
   end