diff --git a/search-service/src/test/kotlin/com/egm/stellio/search/service/AggregatedQueryServiceTests.kt b/search-service/src/test/kotlin/com/egm/stellio/search/service/AggregatedQueryServiceTests.kt
index ea80463b1..3510dce8e 100644
--- a/search-service/src/test/kotlin/com/egm/stellio/search/service/AggregatedQueryServiceTests.kt
+++ b/search-service/src/test/kotlin/com/egm/stellio/search/service/AggregatedQueryServiceTests.kt
@@ -395,7 +395,13 @@ class AggregatedQueryServiceTests : WithTimescaleContainer, WithKafkaContainer {
         }
 
         val temporalEntitiesQuery = createTemporalEntitiesQuery("avg", aggrPeriodDuration)
-        attributeInstanceService.search(temporalEntitiesQuery, temporalEntityAttribute, startTimestamp)
+        attributeInstanceService.search(
+            temporalEntitiesQuery.copy(
+                temporalQuery = temporalEntitiesQuery.temporalQuery.copy(timeAt = startTimestamp)
+            ),
+            temporalEntityAttribute,
+            startTimestamp
+        )
             .shouldSucceedWith { results ->
                 assertEquals(expectedNumberOfBuckets, results.size)
             }