diff --git a/ttnn/CMakeLists.txt b/ttnn/CMakeLists.txt
index c34b042b729..8837211ee45 100644
--- a/ttnn/CMakeLists.txt
+++ b/ttnn/CMakeLists.txt
@@ -734,6 +734,7 @@ function(add_ttnn_sublibrary SUBLIBRARY_NAME)
     else()
         add_library(${SUBLIBRARY_NAME} OBJECT ${ARGN})
     endif()
+    TT_ENABLE_UNITY_BUILD(${SUBLIBRARY_NAME})
     if(WITH_PYTHON_BINDINGS)
         target_compile_definitions(${SUBLIBRARY_NAME} PUBLIC TTNN_WITH_PYTHON_BINDINGS=1)
     else()
diff --git a/ttnn/cpp/ttnn/operations/transformer/concatenate_heads/concatenate_heads.hpp b/ttnn/cpp/ttnn/operations/transformer/concatenate_heads/concatenate_heads.hpp
index e19e61fb885..dc5842c28e2 100644
--- a/ttnn/cpp/ttnn/operations/transformer/concatenate_heads/concatenate_heads.hpp
+++ b/ttnn/cpp/ttnn/operations/transformer/concatenate_heads/concatenate_heads.hpp
@@ -2,6 +2,8 @@
 //
 // SPDX-License-Identifier: Apache-2.0
 
+#pragma once
+
 #include "ttnn/decorators.hpp"
 
 namespace ttnn {