diff --git a/docs/make.jl b/docs/make.jl
index 7307e9393..c2e3db45d 100644
--- a/docs/make.jl
+++ b/docs/make.jl
@@ -13,6 +13,7 @@ makedocs(;
     sitename = "PromptingTools.jl",
     format = Documenter.HTML(;
         prettyurls = get(ENV, "CI", "false") == "true",
+        repolink = "https://github.com/svilupp/PromptingTools.jl",
         canonical = "https://svilupp.github.io/PromptingTools.jl",
         edit_link = "main",
         assets = String[]),
diff --git a/test/llm_openai.jl b/test/llm_openai.jl
index b5234b475..135c14547 100644
--- a/test/llm_openai.jl
+++ b/test/llm_openai.jl
@@ -1,4 +1,5 @@
 using PromptingTools: TestEchoOpenAISchema, render, OpenAISchema
+using PromptingTools: AIMessage, SystemMessage, UserMessage, DataMessage
 
 @testset "render-OpenAI" begin
     schema = OpenAISchema()
diff --git a/test/messages.jl b/test/messages.jl
index 507ff0522..0068712c3 100644
--- a/test/messages.jl
+++ b/test/messages.jl
@@ -1,3 +1,5 @@
+using PromptingTools: AIMessage, SystemMessage, UserMessage, DataMessage
+
 @testset "Message constructors" begin
     # Creates an instance of MSG with the given content string.
     content = "Hello, world!"