diff --git a/tt-train/sources/ttml/CMakeLists.txt b/tt-train/sources/ttml/CMakeLists.txt
index 5e8bdbac7cc..39cc5f7c034 100644
--- a/tt-train/sources/ttml/CMakeLists.txt
+++ b/tt-train/sources/ttml/CMakeLists.txt
@@ -66,9 +66,9 @@ if(NOT TARGET Metalium::Metal)
         message(FATAL_ERROR "libtt_metal.so not found in ${METALIUM_LIB_PATH}")
     endif()
     if(TTNN_LIBRARY)
-        add_library(Metalium::PYBIND SHARED IMPORTED)
+        add_library(Metalium::TTNN SHARED IMPORTED)
         set_target_properties(
-            Metalium::PYBIND
+            Metalium::TTNN
             PROPERTIES
                 IMPORTED_LOCATION
                     "${TTNN_LIBRARY}"
@@ -101,7 +101,7 @@ target_link_libraries(
         pthread
         atomic
         Metalium::Metal
-        Metalium::PYBIND
+        Metalium::TTNN
         Python::Python
         fmt::fmt-header-only
         magic_enum::magic_enum