diff --git a/README.md b/README.md
index 80bed76..72396ae 100644
--- a/README.md
+++ b/README.md
@@ -141,7 +141,7 @@ y : np.ndarray [shape=(# frames,) or (# channels, # frames)], real-valued
       If unspecified, defaults to ``win_length // 4`` (see below)., by default None
   n_jobs : int, optional
       Number of parallel jobs to run. Set at -1 to use all CPU cores, by default 1
-  torch_flag: bool, optional
+  use_torch: bool, optional
       Whether to use the torch version of spectral gating, by default False
   device: str, optional
       A device to run the torch spectral gating on, by default "cuda"