diff --git a/testing/integration/restrict_upgrade_deb_test.go b/testing/integration/restrict_upgrade_deb_test.go
index 8290a4d1563..80ee581eef3 100644
--- a/testing/integration/restrict_upgrade_deb_test.go
+++ b/testing/integration/restrict_upgrade_deb_test.go
@@ -1,3 +1,7 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License 2.0;
+// you may not use this file except in compliance with the Elastic License 2.0.
+
 //go:build integration
 
 package integration
@@ -7,10 +11,11 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/require"
+
 	"github.com/elastic/elastic-agent/internal/pkg/agent/application/coordinator"
 	atesting "github.com/elastic/elastic-agent/pkg/testing"
 	"github.com/elastic/elastic-agent/pkg/testing/define"
-	"github.com/stretchr/testify/require"
 )
 
 func TestRestrictUpgradeDeb(t *testing.T) {
diff --git a/testing/integration/restrict_upgrade_rpm_test.go b/testing/integration/restrict_upgrade_rpm_test.go
index 293acb08efe..5b1d8e31607 100644
--- a/testing/integration/restrict_upgrade_rpm_test.go
+++ b/testing/integration/restrict_upgrade_rpm_test.go
@@ -1,3 +1,7 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License 2.0;
+// you may not use this file except in compliance with the Elastic License 2.0.
+
 //go:build integration
 
 package integration
@@ -7,10 +11,11 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/require"
+
 	"github.com/elastic/elastic-agent/internal/pkg/agent/application/coordinator"
 	atesting "github.com/elastic/elastic-agent/pkg/testing"
 	"github.com/elastic/elastic-agent/pkg/testing/define"
-	"github.com/stretchr/testify/require"
 )
 
 func TestRestrictUpgradeRPM(t *testing.T) {