diff --git a/kstreamplify-core/src/main/java/com/michelin/kstreamplify/services/ProbeService.java b/kstreamplify-core/src/main/java/com/michelin/kstreamplify/services/ProbeService.java
index a52111f7..283a143e 100644
--- a/kstreamplify-core/src/main/java/com/michelin/kstreamplify/services/ProbeService.java
+++ b/kstreamplify-core/src/main/java/com/michelin/kstreamplify/services/ProbeService.java
@@ -49,7 +49,7 @@ public static RestServiceResponse<String> readinessProbe(
                 }
             }
 
-            return kafkaStreamsInitializer.getKafkaStreams().state().isRunningOrRebalancing()
+            return kafkaStreamsInitializer.getKafkaStreams().state().equals(KafkaStreams.State.RUNNING)
                 ? RestServiceResponse.<String>builder().status(HttpURLConnection.HTTP_OK).build() :
                 RestServiceResponse.<String>builder().status(HttpURLConnection.HTTP_UNAVAILABLE)
                     .build();