diff --git a/main_test.go b/main_test.go
index e31be08..120fc37 100644
--- a/main_test.go
+++ b/main_test.go
@@ -3,7 +3,7 @@ package main_test
 import (
 	"testing"
 
-	"github.com/aws-samples/lambda-go-samples"
+	main "github.com/aws-samples/lambda-go-samples"
 
 	"github.com/aws/aws-lambda-go/events"
 	"github.com/stretchr/testify/assert"