diff --git a/cli/src/commands/state/util.rs b/cli/src/commands/state/util.rs
index 5f0b67ce1..96d4c9d05 100644
--- a/cli/src/commands/state/util.rs
+++ b/cli/src/commands/state/util.rs
@@ -1,3 +1,13 @@
+// Copyright (c) 2024 - Restate Software, Inc., Restate GmbH.
+// All rights reserved.
+//
+// Use of this software is governed by the Business Source License
+// included in the LICENSE file.
+//
+// As of the Change Date specified in that file, in accordance with
+// the Business Source License, use of this software will be governed
+// by the Apache License, Version 2.0.
+
 use crate::cli_env::CliEnv;
 use crate::clients::{MetaClientInterface, MetasClient};
 use crate::ui::console::StyledTable;
diff --git a/cli/src/ui/output.rs b/cli/src/ui/output.rs
index 6dbf9dd6e..de20e06bf 100644
--- a/cli/src/ui/output.rs
+++ b/cli/src/ui/output.rs
@@ -1,3 +1,4 @@
+// Copyright (c) 2024 - Restate Software, Inc., Restate GmbH.
 // All rights reserved.
 //
 // Use of this software is governed by the Business Source License
diff --git a/crates/benchmarks/src/lib.rs b/crates/benchmarks/src/lib.rs
index 6753c56e1..25fa196c2 100644
--- a/crates/benchmarks/src/lib.rs
+++ b/crates/benchmarks/src/lib.rs
@@ -1,4 +1,3 @@
-#![allow(clippy::async_yields_async)]
 // Copyright (c) 2023 -  Restate Software, Inc., Restate GmbH.
 // All rights reserved.
 //
@@ -9,6 +8,8 @@
 // the Business Source License, use of this software will be governed
 // by the Apache License, Version 2.0.
 
+#![allow(clippy::async_yields_async)]
+
 //! Utilities for benchmarking the Restate runtime
 use std::num::NonZeroU64;
 use std::time::Duration;
diff --git a/crates/service-client/src/request_identity/mod.rs b/crates/service-client/src/request_identity/mod.rs
index cfecbc4e1..090596441 100644
--- a/crates/service-client/src/request_identity/mod.rs
+++ b/crates/service-client/src/request_identity/mod.rs
@@ -1,3 +1,13 @@
+// Copyright (c) 2024 - Restate Software, Inc., Restate GmbH.
+// All rights reserved.
+//
+// Use of this software is governed by the Business Source License
+// included in the LICENSE file.
+//
+// As of the Change Date specified in that file, in accordance with
+// the Business Source License, use of this software will be governed
+// by the Apache License, Version 2.0.
+
 use hyper::header::HeaderName;
 
 use hyper::HeaderMap;
diff --git a/crates/service-client/src/request_identity/v1.rs b/crates/service-client/src/request_identity/v1.rs
index d05169ec1..15529ae43 100644
--- a/crates/service-client/src/request_identity/v1.rs
+++ b/crates/service-client/src/request_identity/v1.rs
@@ -1,3 +1,13 @@
+// Copyright (c) 2024 - Restate Software, Inc., Restate GmbH.
+// All rights reserved.
+//
+// Use of this software is governed by the Business Source License
+// included in the LICENSE file.
+//
+// As of the Change Date specified in that file, in accordance with
+// the Business Source License, use of this software will be governed
+// by the Apache License, Version 2.0.
+
 use std::fmt;
 use std::fmt::{Debug, Formatter};
 use std::path::PathBuf;
diff --git a/crates/worker/src/partition/state_machine/command_interpreter/tests.rs b/crates/worker/src/partition/state_machine/command_interpreter/tests.rs
index 2cc5974c1..8278ef5d1 100644
--- a/crates/worker/src/partition/state_machine/command_interpreter/tests.rs
+++ b/crates/worker/src/partition/state_machine/command_interpreter/tests.rs
@@ -1,3 +1,13 @@
+// Copyright (c) 2024 - Restate Software, Inc., Restate GmbH.
+// All rights reserved.
+//
+// Use of this software is governed by the Business Source License
+// included in the LICENSE file.
+//
+// As of the Change Date specified in that file, in accordance with
+// the Business Source License, use of this software will be governed
+// by the Apache License, Version 2.0.
+
 use super::*;
 
 use bytestring::ByteString;
diff --git a/tools/scripts/check-license-headers b/tools/scripts/check-license-headers
index bdc665107..d94887125 100755
--- a/tools/scripts/check-license-headers
+++ b/tools/scripts/check-license-headers
@@ -1,8 +1,9 @@
 #!/usr/bin/env bash
 set -euo pipefail
 exit_code=0
-for file in $(find src -name "*.rs"); do
-    if ! head -9 "$file" | tr -d '\n' | grep -q '^// Copyright (c) [0-9]\{4\}\(-[0-9]\{4\}\)\? -  Restate Software, Inc., Restate GmbH.// All rights reserved.//// Use of this software is governed by the Business Source License// included in the LICENSE file.//// As of the Change Date specified in that file, in accordance with// the Business Source License, use of this software will be governed// by the Apache License, Version 2.0.$'; then
+files=$(find crates cli server -name "*.rs") || exit 1
+for file in ${files}; do
+    if ! head -9 "$file" | tr -d '\n' | grep -q '^// Copyright (c) [0-9]\{4\}\(-[0-9]\{4\}\)\? - \+Restate Software, Inc., Restate GmbH.// All rights reserved.//// Use of this software is governed by the Business Source License// included in the LICENSE file.//// As of the Change Date specified in that file, in accordance with// the Business Source License, use of this software will be governed// by the Apache License, Version 2.0.$'; then
         echo "$file does not contain valid license header"
         exit_code=1
     fi