diff --git a/lib/contexted/module_analyzer.ex b/lib/contexted/module_analyzer.ex
index 61aca68..e812402 100644
--- a/lib/contexted/module_analyzer.ex
+++ b/lib/contexted/module_analyzer.ex
@@ -107,10 +107,10 @@ defmodule Contexted.ModuleAnalyzer do
     Enum.map_join(types, " | ", &format_type/1)
   end
 
-  defp format_type({:atom, _, atom}), do: atom |> Atom.to_string()
+  defp format_type({:atom, _, atom}), do: ":#{Atom.to_string(atom)}"
   defp format_type({:type, _, type_name, _}), do: "#{type_name}()"
 
-  defp format_type({:remote_type, _, [{:atom, _, module}, {:atom, _, type}, []]}) do
+  defp format_type({:remote_type, _, [{:atom, _, module}, {:atom, _, type}, _list]}) do
     if module == :elixir do
       "#{type}()"
     else