diff --git a/test/MetaAugment/test_aa_learner.py b/test/MetaAugment/test_aa_learner.py
index 08092d7a420fff25b7a289e2cc8ee2d88a94565a..3e2808702a04746e625acd5b463cfe01f56687bd 100644
--- a/test/MetaAugment/test_aa_learner.py
+++ b/test/MetaAugment/test_aa_learner.py
@@ -26,8 +26,8 @@ def test_translate_operation_tensor():
         softmax = torch.nn.Softmax(dim=0)
 
         fun_num = random.randint(1, 14)
-        p_bins = random.randint(1, 15)
-        m_bins = random.randint(1, 15)
+        p_bins = random.randint(2, 15)
+        m_bins = random.randint(2, 15)
 
         agent = aal.aa_learner(
                 sp_num=5,
diff --git a/test/MetaAugment/test_gru_learner.py b/test/MetaAugment/test_gru_learner.py
index 2ecb6fbddca436d660133a6b2e6e0250d0cac7da..52c2b8a2a17e87710bddeda3ba50ced0bfc226fe 100644
--- a/test/MetaAugment/test_gru_learner.py
+++ b/test/MetaAugment/test_gru_learner.py
@@ -15,8 +15,8 @@ def test_generate_new_policy():
     for _ in range(40):
         sp_num = random.randint(1,20)
         fun_num = random.randint(1, 14)
-        p_bins = random.randint(1, 15)
-        m_bins = random.randint(1, 15)
+        p_bins = random.randint(2, 15)
+        m_bins = random.randint(2, 15)
 
         agent = aal.gru_learner(
             sp_num=sp_num,
@@ -24,7 +24,7 @@ def test_generate_new_policy():
             p_bins=p_bins,
             m_bins=m_bins
             )
-        for _ in range(10):
+        for _ in range(4):
             new_policy = agent.generate_new_policy()
             assert isinstance(new_policy[0], list), new_policy
 
diff --git a/test/MetaAugment/test_randomsearch_learner.py b/test/MetaAugment/test_randomsearch_learner.py
index 83ddfef1a0a192d279967ce799d12116da574139..5b67d98e1f2e40d56b3aac2445f041f1372bbe9f 100644
--- a/test/MetaAugment/test_randomsearch_learner.py
+++ b/test/MetaAugment/test_randomsearch_learner.py
@@ -17,8 +17,8 @@ def test_generate_new_policy():
         for _ in range(40):
             sp_num = random.randint(1,20)
             fun_num = random.randint(1, 14)
-            p_bins = random.randint(1, 15)
-            m_bins = random.randint(1, 15)
+            p_bins = random.randint(2, 15)
+            m_bins = random.randint(2, 15)
 
             agent = aal.randomsearch_learner(
                 sp_num=sp_num,
@@ -27,7 +27,7 @@ def test_generate_new_policy():
                 m_bins=m_bins,
                 discrete_p_m=discrete_p_m
                 )
-            for _ in range(10):
+            for _ in range(4):
                 new_policy = agent.generate_new_policy()
                 assert isinstance(new_policy, list), new_policy
     
diff --git a/test/README.md b/test/README.md
index 895175a2d867ffad637d5690952bda9db76ba92e..27f6a4eb8e9409f84e66a65bad0708f098310bab 100644
--- a/test/README.md
+++ b/test/README.md
@@ -1,4 +1,6 @@
-This directory contatins all unit tests and integration tests for this project,
-including for the library and the web-app.
+This directory contains all unit tests and integration tests for this project.
+
+The tests for the library can be seen in /test/MetaAugment.
+The tests for the web-app can be seen in ????
 
 Use command `pytest` in main directory to run all tests in this directory.
\ No newline at end of file