diff --git a/python/graph_index.cpp b/python/graph_index.cpp
index 6396247..d37db03 100644
--- a/python/graph_index.cpp
+++ b/python/graph_index.cpp
@@ -53,7 +53,7 @@ struct GraphIndex{
     }
 
     NeighborsAndDistances batch_search(py::array_t<T, py::array::c_style | py::array::forcecast> &queries, uint64_t num_queries, uint64_t knn,
-                        uint64_t beam_width, uint64_t visit_limit = -1){
+                        uint64_t beam_width, int64_t visit_limit = -1){
         if(visit_limit == -1) visit_limit = G.size();
         QueryParams QP(knn, beam_width, 1.35, visit_limit, G.max_degree());