diff --git a/sw/test/test_proximal.py b/sw/test/test_proximal.py
index 355efc6..650b0e6 100644
--- a/sw/test/test_proximal.py
+++ b/sw/test/test_proximal.py
@@ -20,7 +20,7 @@ class CheckParityTestCase(unittest.TestCase):
                       [0, 0, 0, 0, 0, 1, 0],
                       [0, 0, 0, 0, 0, 0, 1]])
 
-        decoder = proximal.ProximalDecoder(H, R)
+        decoder = proximal.ProximalDecoder(H)
 
         d1 = np.array([0, 1, 0, 1])
         c1 = np.dot(np.transpose(G), d1) % 2
@@ -62,7 +62,7 @@ class GradientTestCase(unittest.TestCase):
         expected_grad_h = np.array(
             [4, 26, -8, -36, 38, 28, -32])  # Manually calculated result
 
-        decoder = proximal.ProximalDecoder(H, R)
+        decoder = proximal.ProximalDecoder(H)
         grad_h = decoder._grad_h(x)
 
         self.assertEqual(np.array_equal(grad_h, expected_grad_h), True)