diff --git a/examples/llama.android/lib/src/main/cpp/CMakeLists.txt b/examples/llama.android/lib/src/main/cpp/CMakeLists.txt
index c7fd8d63cf..06d3d03aa5 100644
--- a/examples/llama.android/lib/src/main/cpp/CMakeLists.txt
+++ b/examples/llama.android/lib/src/main/cpp/CMakeLists.txt
@@ -24,7 +24,7 @@ target_link_libraries(cpu-detector
         log)
 
 # --------------------------------------------------------------------------
-# 2. Kleidi Llama library
+# 2. AI Chat library
 # --------------------------------------------------------------------------
 
 if(DEFINED ANDROID_ABI)
@@ -46,7 +46,7 @@ set(LLAMA_SRC ${CMAKE_CURRENT_LIST_DIR}/../../../../../../)
 add_subdirectory(${LLAMA_SRC} build-llama)
 
 add_library(${CMAKE_PROJECT_NAME} SHARED
-        kleidi_llama.cpp)
+        ai_chat.cpp)
 
 target_compile_definitions(${CMAKE_PROJECT_NAME} PRIVATE
         GGML_SYSTEM_ARCH=${GGML_SYSTEM_ARCH}
diff --git a/examples/llama.android/lib/src/main/cpp/kleidi_llama.cpp b/examples/llama.android/lib/src/main/cpp/ai_chat.cpp
similarity index 100%
rename from examples/llama.android/lib/src/main/cpp/kleidi_llama.cpp
rename to examples/llama.android/lib/src/main/cpp/ai_chat.cpp
diff --git a/examples/llama.android/lib/src/main/java/com/arm/aichat/internal/InferenceEngineImpl.kt b/examples/llama.android/lib/src/main/java/com/arm/aichat/internal/InferenceEngineImpl.kt
index a22c5b7620..783f8b31bb 100644
--- a/examples/llama.android/lib/src/main/java/com/arm/aichat/internal/InferenceEngineImpl.kt
+++ b/examples/llama.android/lib/src/main/java/com/arm/aichat/internal/InferenceEngineImpl.kt
@@ -39,7 +39,7 @@ import java.io.IOException
  *
  * State transitions are managed automatically and validated at each operation.
  *
- * @see kleidi-llama.cpp for the native implementation details
+ * @see ai_chat.cpp for the native implementation details
  */
 internal class InferenceEngineImpl private constructor(
     private val nativeLibDir: String
@@ -75,7 +75,7 @@ internal class InferenceEngineImpl private constructor(
 
     /**
      * JNI methods
-     * @see kleidi-llama.cpp
+     * @see ai_chat.cpp
      */
     @FastNative
     private external fun init(nativeLibDir: String)