diff --git a/sw/decoders/proximal.py b/sw/decoders/proximal.py
index eff012a..5d1765e 100644
--- a/sw/decoders/proximal.py
+++ b/sw/decoders/proximal.py
@@ -6,19 +6,19 @@ class ProximalDecoder:
     by Tadashi Wadayama, and Satoshi Takabe.
     """
 
-    def __init__(self, H: np.array, K: int = 100, step_size: float = 0.1,
+    def __init__(self, H: np.array, K: int = 100, omega: float = 0.1,
                  gamma: float = 0.05, eta: float = 1.5):
         """Construct a new ProximalDecoder Object.
 
         :param H: Parity Check Matrix
         :param K: Max number of iterations to perform when decoding
-        :param step_size: Step size for the gradient descent process
+        :param omega: Step size for the gradient descent process
         :param gamma: Positive constant. Arises in the approximation of the prior PDF
         :param eta: Positive constant slightly larger than one. See 3.2,  p. 3
         """
         self._H = H
         self._K = K
-        self._step_size = step_size
+        self._step_size = omega
         self._gamma = gamma
         self._eta = eta