diff --git a/.licensei.toml b/.licensei.toml
index 0f69dba..5c1d8d4 100644
--- a/.licensei.toml
+++ b/.licensei.toml
@@ -7,4 +7,27 @@ approved = [
 ]
 
 ignored = [
+  "google.golang.org/protobuf",
 ]
+
+[header]
+ignoreFiles = [
+  "allspark.pb.go",
+  "producer.go",
+  "kafkaProduce.go",
+  "kafkaConsume.go",
+  "consumer.go",
+]
+template = """// Copyright © :YEAR: Banzai Cloud
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License."""
diff --git a/Makefile b/Makefile
index c9c595c..f7560e0 100644
--- a/Makefile
+++ b/Makefile
@@ -28,7 +28,7 @@ DOCKER_TAG ?= ${VERSION}
 # Dependency versions
 GOLANGCI_VERSION = 1.45.0
 GOLANG_VERSION = 1.18
-LICENSEI_VERSION = 0.3.1
+LICENSEI_VERSION = 0.6.1
 
 # Add the ability to override some variables
 # Use with care
@@ -122,7 +122,7 @@ bin/licensei-${LICENSEI_VERSION}:
 .PHONY: license-check
 license-check: bin/licensei ## Run license check
 	bin/licensei check
-	./scripts/check-header.sh
+	bin/licensei header
 
 .PHONY: license-cache
 license-cache: bin/licensei ## Generate license cache