diff --git a/pkg/system/host_data_provider_test.go b/pkg/system/host_data_provider_test.go
index ce42d3f..e73ac7c 100644
--- a/pkg/system/host_data_provider_test.go
+++ b/pkg/system/host_data_provider_test.go
@@ -1,17 +1,18 @@
 package system_test
 
 import (
-	"github.com/pantheon-systems/pauditd/pkg/system"
 	"os"
 	"testing"
 
+	"github.com/pantheon-systems/pauditd/pkg/system"
+
 	"github.com/stretchr/testify/assert"
 )
 
 func Test_GetHostnameWithEnvVar(t *testing.T) {
 	testName := "test-node-name"
 
-	os.Setenv("NODENAME", testName)
+	os.Setenv("HOSTNAME", testName)
 
 	hostname := system.GetHostname()