diff --git a/sw/test/test_proximal.py b/sw/test/test_proximal.py
index ff49684..ed3afb8 100644
--- a/sw/test/test_proximal.py
+++ b/sw/test/test_proximal.py
@@ -37,9 +37,9 @@ class CheckParityTestCase(unittest.TestCase):
 class GradientTestCase(unittest.TestCase):
     """Test case for the calculation of the gradient of the code-constraint-polynomial"""
     def test_grad_h(self):
-        H = np.array([[1, 0, 1],
+        H = np.array([[1, 0, 0],
                       [0, 1, 0]])
-        x = np.array([2, 3, 4])
+        x = np.array([1, 2, 2])
 
         decoder = proximal.ProximalDecoder(H)
         grad = decoder._grad_h(x)