diff --git a/tests/test_gpu_aamp_ostinato.py b/tests/test_gpu_aamp_ostinato.py
index 4545ae48c..8284aeff2 100644
--- a/tests/test_gpu_aamp_ostinato.py
+++ b/tests/test_gpu_aamp_ostinato.py
@@ -80,13 +80,13 @@ def test_input_not_overwritten():
 def test_extract_several_consensus():
     # This test is to further ensure that the function `gpu_aamp_ostinato`
     # does not tamper with the original data.
-    Ts = [np.random.rand(n) for n in [256, 512, 1024]]
+    Ts = [np.random.rand(n) for n in [64, 128]]
     Ts_ref = [T.copy() for T in Ts]
     Ts_comp = [T.copy() for T in Ts]
 
     m = 20
 
-    k = 5  # Get the first `k` consensus motifs
+    k = 2  # Get the first `k` consensus motifs
     for _ in range(k):
         # Find consensus motif and its NN in each time series in Ts_comp
         # Remove them from Ts_comp as well as Ts_ref, and assert that the
diff --git a/tests/test_gpu_ostinato.py b/tests/test_gpu_ostinato.py
index 9bd8f86cf..30eb9539c 100644
--- a/tests/test_gpu_ostinato.py
+++ b/tests/test_gpu_ostinato.py
@@ -141,13 +141,13 @@ def test_input_not_overwritten():
 def test_extract_several_consensus():
     # This test is to further ensure that the function `gpu_ostinato`
     # does not tamper with the original data.
-    Ts = [np.random.rand(n) for n in [256, 512, 1024]]
+    Ts = [np.random.rand(n) for n in [64, 128]]
     Ts_ref = [T.copy() for T in Ts]
     Ts_comp = [T.copy() for T in Ts]
 
     m = 20
 
-    k = 5  # Get the first `k` consensus motifs
+    k = 2  # Get the first `k` consensus motifs
     for _ in range(k):
         # Find consensus motif and its NN in each time series in Ts_comp
         # Remove them from Ts_comp as well as Ts_ref, and assert that the