diff --git a/diskSubHeader_whitebox_test.go b/diskSubHeader_whitebox_test.go
index a024331..4f8a398 100644
--- a/diskSubHeader_whitebox_test.go
+++ b/diskSubHeader_whitebox_test.go
@@ -1,4 +1,4 @@
-package main
+package goirsdk
 
 import (
 	"testing"
diff --git a/headers_whitebox_test.go b/headers_whitebox_test.go
index a767268..c2d24f3 100644
--- a/headers_whitebox_test.go
+++ b/headers_whitebox_test.go
@@ -1,4 +1,4 @@
-package main
+package goirsdk
 
 import (
 	"testing"