diff --git a/src/neuron_proofreader/geometric_learning/curve_visualization.py b/src/neuron_proofreader/geometric_learning/curve_visualization.py
index 24db69f..116df54 100644
--- a/src/neuron_proofreader/geometric_learning/curve_visualization.py
+++ b/src/neuron_proofreader/geometric_learning/curve_visualization.py
@@ -67,7 +67,7 @@ def plot_length_distribution(dataset, output_path=None, title=None):
     # Compute path length stats
     lengths = dataset.examples_df["length"].to_numpy()
     p50, p99 = np.percentile(lengths, [50, 99.9])
-    lengths = lengths[lengths <= p999]
+    lengths = lengths[lengths <= p99]
 
     # Plot path lengths
     plt.figure(figsize=(8, 5))