diff --git a/parsemail_test.go b/parsemail_test.go
index 6f5b52b..fb9a231 100644
--- a/parsemail_test.go
+++ b/parsemail_test.go
@@ -1,8 +1,7 @@
-package parsemail_test
+package parsemail
 
 import (
 	"encoding/base64"
-	"github.com/DusanKasan/parsemail"
 	"io/ioutil"
 	"net/mail"
 	"strings"
@@ -180,7 +179,7 @@
 	}
 
 	for index, td := range testData {
-		e, err := parsemail.Parse(strings.NewReader(td.mailData))
+		e, err := Parse(strings.NewReader(td.mailData))
 		if err != nil {
 			t.Error(err)
 		}