diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultsTestingRepositoryImpl.java b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultsTestingRepositoryImpl.java
index 2964c43d..fbd0a60e 100644
--- a/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultsTestingRepositoryImpl.java
+++ b/src/main/java/com/kamco/cd/kamcoback/postgres/repository/Inference/InferenceResultsTestingRepositoryImpl.java
@@ -23,11 +23,12 @@ public class InferenceResultsTestingRepositoryImpl
         .select(inferenceResultsTestingEntity)
         .from(inferenceResultsTestingEntity)
         .where(
-            inferenceResultsTestingEntity
-                .batchId
-                .in(batchIds)
-                .and(inferenceResultsTestingEntity.afterC.isNotNull())
-                .and(inferenceResultsTestingEntity.afterP.isNotNull()))
+            inferenceResultsTestingEntity.batchId.in(batchIds),
+            inferenceResultsTestingEntity.afterC.isNotNull(),
+            inferenceResultsTestingEntity.afterP.isNotNull(),
+            inferenceResultsTestingEntity.beforeC.isNotNull(),
+            inferenceResultsTestingEntity.beforeP.isNotNull(),
+            inferenceResultsTestingEntity.probability.isNotNull())
         .fetch();
   }
 
@@ -44,7 +45,10 @@ public class InferenceResultsTestingRepositoryImpl
             .where(
                 inferenceResultsTestingEntity.batchId.in(batchIds),
                 inferenceResultsTestingEntity.afterC.isNotNull(),
-                inferenceResultsTestingEntity.afterP.isNotNull())
+                inferenceResultsTestingEntity.afterP.isNotNull(),
+                inferenceResultsTestingEntity.beforeC.isNotNull(),
+                inferenceResultsTestingEntity.beforeP.isNotNull(),
+                inferenceResultsTestingEntity.probability.isNotNull())
             .fetchOne();
 
     return cnt == null ? 0L : cnt;
@@ -62,11 +66,12 @@ public class InferenceResultsTestingRepositoryImpl
                 inferenceResultsTestingEntity.afterYear.max()))
         .from(inferenceResultsTestingEntity)
         .where(
-            inferenceResultsTestingEntity
-                .batchId
-                .in(batchIds)
-                .and(inferenceResultsTestingEntity.afterC.isNotNull())
-                .and(inferenceResultsTestingEntity.afterP.isNotNull()))
+            inferenceResultsTestingEntity.batchId.in(batchIds),
+            inferenceResultsTestingEntity.afterC.isNotNull(),
+            inferenceResultsTestingEntity.afterP.isNotNull(),
+            inferenceResultsTestingEntity.beforeC.isNotNull(),
+            inferenceResultsTestingEntity.beforeP.isNotNull(),
+            inferenceResultsTestingEntity.probability.isNotNull())
         .groupBy(
             inferenceResultsTestingEntity.batchId,
             inferenceResultsTestingEntity.modelVersion,