diff --git a/diskann/src/lib.rs b/diskann/src/lib.rs
index f3dede8..f795747 100644
--- a/diskann/src/lib.rs
+++ b/diskann/src/lib.rs
@@ -55,7 +55,6 @@ impl IndexGraph {
 #[derive(Clone, Copy, Debug)]
 pub struct IndexBuildConfig {
     pub r: usize,
-    pub r_cap: usize,
     pub l: usize,
     pub maxc: usize,
     pub alpha: i64
@@ -354,7 +353,7 @@ pub fn augment_bipartite(rng: &mut Rng, graph: &mut IndexGraph, query_knns: Vec<
     sigmas.into_par_iter().for_each_init(|| rng.lock().unwrap().fork(), |rng, sigma_i| {
         let mut neighbours = graph.out_neighbours_mut(sigma_i);
         let mut i = 0;
-        while neighbours.len() < config.r_cap && i < 100 {
+        while neighbours.len() < config.r && i < 100 {
             let query_neighbour = *rng.choice(&query_knns[sigma_i as usize]).unwrap();
             let projected_neighbour = *rng.choice(&query_knns_bwd[query_neighbour as usize]).unwrap();
             if !neighbours.contains(&projected_neighbour) {
diff --git a/diskann/src/main.rs b/diskann/src/main.rs
index 037b15e..ba3208a 100644
--- a/diskann/src/main.rs
+++ b/diskann/src/main.rs
@@ -60,7 +60,6 @@ fn main() -> Result<()> {
 
         let mut config = IndexBuildConfig {
             r: 64,
-            r_cap: 64,
             l: 192,
             maxc: 750,
             alpha: 65200,
@@ -93,7 +92,6 @@ fn main() -> Result<()> {
 
     let mut config = IndexBuildConfig {
         r: 64,
-        r_cap: 64,
         l: 50,
         alpha: 65536,
         maxc: 0,
diff --git a/src/generate_index_shard.rs b/src/generate_index_shard.rs
index b60b00c..81125b8 100644
--- a/src/generate_index_shard.rs
+++ b/src/generate_index_shard.rs
@@ -42,7 +42,6 @@ fn main() -> Result<()> {
 
     let mut config = IndexBuildConfig {
         r: 64,
-        r_cap: 64,
         l: 200,
         maxc: 750,
         alpha: 65300
@@ -54,7 +53,7 @@ fn main() -> Result<()> {
         length: original_ids.len()
     };
 
-    let mut graph = IndexGraph::empty(original_ids.len(), config.r_cap);
+    let mut graph = IndexGraph::empty(original_ids.len(), config.r);
 
     {
         //let _timer = Timer::new("project bipartite");