diff --git a/pages/repo/html_test.go b/pages/repo/html_test.go
index 13224d3..ebb9c82 100644
--- a/pages/repo/html_test.go
+++ b/pages/repo/html_test.go
@@ -5,14 +5,25 @@ package repo
 
 import (
 	"testing"
+
+	"github.com/pocka/legit/config"
+	"github.com/pocka/legit/core"
 )
 
 func TestRewriteInternalMediaSource(t *testing.T) {
+	var c config.Config
+	c.Repo.ScanPath = t.TempDir()
+	core, err := core.New(&c)
+	if err != nil {
+		t.Fatal(err)
+	}
+
 	repo := Repo{
 		dirname: "foo",
+		core:    core,
 	}
 
-	r := newRepoLinkTransformer(&repo, "trunk")
+	r := newRepoLinkTransformer(&repo, "trunk", "bar.md")
 
 	{
 		input := "media/screenshot.png"
@@ -61,15 +72,23 @@ func TestRewriteInternalMediaSource(t *testing.T) {
 }
 
 func TestRewriteInternalLink(t *testing.T) {
+	var c config.Config
+	c.Repo.ScanPath = t.TempDir()
+	core, err := core.New(&c)
+	if err != nil {
+		t.Fatal(err)
+	}
+
 	repo := Repo{
 		dirname: "foo",
+		core:    core,
 	}
 
-	r := newRepoLinkTransformer(&repo, "trunk")
+	r := newRepoLinkTransformer(&repo, "trunk", "bar.md")
 
 	{
 		input := "CHANGELOG.md"
-		expected := "/foo/blob/trunk/CHANGELOG.md"
+		expected := "/foo/blob/trunk/CHANGELOG.md?preview=html"
 
 		if out := r.RewriteInternalLink(input); out != expected {
 			t.Errorf("Expected %s, got %s", expected, out)