diff --git a/search_test.go b/search_test.go
index f6b196a..99b9f1a 100644
--- a/search_test.go
+++ b/search_test.go
@@ -1,12 +1,10 @@
 // Copyright 2020-21 PJ Engineering and Business Solutions Pty. Ltd. All rights reserved.
 
-package googlesearch_test
+package googlesearch
 
 import (
 	"context"
 	"testing"
-
-	googlesearch "github.com/chrisjoyce911/google-search"
 )
 
 var ctx = context.Background()
@@ -15,11 +13,11 @@ func TestSearch(t *testing.T) {
 
 	q := "Hello World"
 
-	opts := googlesearch.SearchOptions{
+	opts := SearchOptions{
 		Limit: 20,
 	}
 
-	returnLinks, err := googlesearch.Search(ctx, q, opts)
+	returnLinks, err := Search(ctx, q, opts)
 	if err != nil {
 		t.Errorf("something went wrong: %v", err)
 		return