diff --git a/src/main.rs b/src/main.rs
index c3ee9e0..92edc24 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -41,6 +41,9 @@ struct Args {
 
     #[arg(long, default_value = "bench_config.toml")]
     config: String,
+
+    #[arg(long, default_value_t = false)]
+    faucet_only: bool,
 }
 
 // mod uniswap;
@@ -479,6 +482,11 @@ async fn start_bench() -> Result<()> {
         .await?;
     }
 
+    if args.faucet_only {
+        info!("Faucet distribution complete. Faucet-only mode enabled, exiting.");
+        return Ok(());
+    }
+
     let tps = benchmark_config.target_tps as usize;
     let duration_secs = benchmark_config.performance.duration_secs;
     if benchmark_config.enable_swap_token {