diff --git a/html2pdf_test.go b/html2pdf_test.go
index 7290244..fd9aa02 100644
--- a/html2pdf_test.go
+++ b/html2pdf_test.go
@@ -7,6 +7,7 @@ import (
 	"testing"
 
 	"gitea.tecamino.com/paadi/html2pdf/converter"
+	"gitea.tecamino.com/paadi/html2pdf/models"
 )
 
 func TestConvert(t *testing.T) {
@@ -27,14 +28,14 @@ func TestConvertFiles(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	var input []converter.File
+	var input []models.File
 	for _, f := range files {
 		ext := filepath.Ext(f.Name())
 		if ext != ".html" {
 			continue
 		}
 
-		input = append(input, converter.File{
+		input = append(input, models.File{
 			Input:  filepath.Join(rootPath, f.Name()),
 			Output: strings.Replace(f.Name(), ext, ".pdf", 1),
 		})