diff --git a/src/index.cpp b/src/index.cpp
index 4157edcef..9655a074e 100644
--- a/src/index.cpp
+++ b/src/index.cpp
@@ -2251,14 +2251,13 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::search(const T *query, con
             break;
     }
 
-    if (pos < K)
+    if (pos <= K)
     {
-        context.SetState(State::Failure);
-        diskann::cerr << "Found pos: " << pos << "fewer than K elements " << K << " for query" << std::endl;
+        context.SetState(State::Success);
     }
     else
     {
-        context.SetState(State::Success);
+        context.SetState(State::Failure);
     }
 
     return retval;
@@ -2387,14 +2386,13 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::search_with_filters(const
         if (pos == K)
             break;
     }
-    if (pos < K)
+    if (pos <= K)
     {
-        context.SetState(State::Failure);
-        diskann::cerr << "Found fewer than K elements for query" << std::endl;
+        context.SetState(State::Success);
     }
     else
     {
-        context.SetState(State::Success);
+        context.SetState(State::Failure);
     }
 
     return retval;