diff --git a/codeflash/languages/java/instrumentation.py b/codeflash/languages/java/instrumentation.py
index 1cacbef5b..5c3a33d4f 100644
--- a/codeflash/languages/java/instrumentation.py
+++ b/codeflash/languages/java/instrumentation.py
@@ -315,7 +315,7 @@ def _collect_calls(
 def _byte_to_line_index(byte_offset: int, line_byte_starts: list[int]) -> int:
     """Map a byte offset in body_text to a body_lines index."""
     idx = bisect.bisect_right(line_byte_starts, byte_offset) - 1
-    return max(idx, 0)
+    return 0 if idx < 0 else idx
 
 
 def _infer_array_cast_type(line: str) -> str | None: