diff --git a/tests/test_activation.py b/tests/test_activation.py
index 848db0b..f3163b3 100644
--- a/tests/test_activation.py
+++ b/tests/test_activation.py
@@ -46,7 +46,7 @@ def test_activation_matrix():
               (['c1', 'c3'], []),
               (['c2'], []),
               (['c1', 'c1'], [])]
-    reference_activations = np.array([[1, 1], [0, 1], [1, 0], [0, 1]]).T
+    reference_activations = np.array([[1, 0, 1, 0], [1, 1, 0, 1]])
 
     with pytest.raises(ValueError):
         activations = activation(events, weights, number_of_threads=1)
@@ -70,7 +70,7 @@ def test_ignore_missing_cues():
               (['c1', 'c3'], []),
               (['c2', 'c4'], []),
               (['c1', 'c1'], [])]
-    reference_activations = np.array([[1, 1], [0, 1], [1, 0], [0, 1]]).T
+    reference_activations = np.array([[1, 0, 1, 0], [1, 1, 0, 1]])
 
     with pytest.raises(KeyError):
         activations = activation(events, weights, number_of_threads=1,