diff --git a/src/bin/generate_secret.rs b/src/bin/generate_secret.rs
index 703637a..80d87a0 100644
--- a/src/bin/generate_secret.rs
+++ b/src/bin/generate_secret.rs
@@ -1,7 +1,10 @@
 use base64::{prelude::BASE64_URL_SAFE_NO_PAD, Engine};
 
 fn main() {
-    let mut key = [0u8; 32];
+    let mut arguments = std::env::args();
+    _ = arguments.next();
+    let argument = arguments.next().unwrap();
+    let mut key = vec![0u8; argument.parse().unwrap()];
     getrandom::getrandom(&mut key).unwrap();
     let encoded = BASE64_URL_SAFE_NO_PAD.encode(&key);
     println!("{}", encoded);