diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultRepositoryImpl.java b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultRepositoryImpl.java
index 9b623183..755bf60e 100644
--- a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultRepositoryImpl.java
+++ b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultRepositoryImpl.java
@@ -184,9 +184,9 @@ public class InferenceResultRepositoryImpl implements InferenceResultRepositoryC
             SELECT
                 r.uid AS result_uid,
                 msadi.stage,
-                r.cd_prob,
-                r.input1 AS compare_yyyy,
-                r.input2 AS target_yyyy,
+                r.after_p as cd_prob,
+                msl.compare_yyyy,
+                msl.target_yyyy,
                 CASE
                   WHEN r.map_id ~ '^[0-9]+$' THEN r.map_id::bigint
                   ELSE NULL