diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 4d0c201350d0..efda5704a7c9 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -461,6 +461,7 @@ if(USE_AI)
 
   FILE(GLOB SOURCE_FILES_AI
     "common/ai_models.c"
+    "common/ai/segmentation.c"
     "develop/masks/object.c"
     "gui/preferences_ai.c"
   )
diff --git a/src/ai/CMakeLists.txt b/src/ai/CMakeLists.txt
index 4d343e0f54d4..44a9bfa1c662 100644
--- a/src/ai/CMakeLists.txt
+++ b/src/ai/CMakeLists.txt
@@ -4,8 +4,6 @@ add_library(darktable_ai STATIC
   backend.h
   backend_common.c
   backend_onnx.c
-  segmentation.h
-  segmentation.c
 )
 
 # Find ONNX Runtime (auto-downloads if not present)
diff --git a/src/ai/segmentation.c b/src/common/ai/segmentation.c
similarity index 99%
rename from src/ai/segmentation.c
rename to src/common/ai/segmentation.c
index b20ed2524640..40c7aed8c9c4 100644
--- a/src/ai/segmentation.c
+++ b/src/common/ai/segmentation.c
@@ -16,8 +16,8 @@
     along with darktable.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#include "segmentation.h"
-#include "backend.h"
+#include "common/ai/segmentation.h"
+#include "ai/backend.h"
 #include "common/darktable.h"
 #include <inttypes.h>
 #include <math.h>
diff --git a/src/ai/segmentation.h b/src/common/ai/segmentation.h
similarity index 99%
rename from src/ai/segmentation.h
rename to src/common/ai/segmentation.h
index 05b92c7fa6c0..c82619092d05 100644
--- a/src/ai/segmentation.h
+++ b/src/common/ai/segmentation.h
@@ -18,7 +18,7 @@
 
 #pragma once
 
-#include "backend.h"
+#include "ai/backend.h"
 #include <glib.h>
 
 /**
diff --git a/src/develop/masks/object.c b/src/develop/masks/object.c
index 94df91ba6b81..20d6f21f7dbf 100644
--- a/src/develop/masks/object.c
+++ b/src/develop/masks/object.c
@@ -16,7 +16,7 @@
     along with darktable.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-#include "ai/segmentation.h"
+#include "common/ai/segmentation.h"
 #include "common/ai_models.h"
 #include "common/colorspaces.h"
 #include "common/debug.h"