diff --git a/makefile b/makefile
index 8a1eb51..ba5e28b 100644
--- a/makefile
+++ b/makefile
@@ -4,7 +4,7 @@ lint:
 	flake8 --ignore E501,W503 .
 
 test:
-	python main.py --alpha 2 --lr 2e-4 --max_epochs 200
+	python main.py --alpha 2 --lr 2e-4 --max_epochs 200 --bs 8192
 
 search:
 	python search.py
diff --git a/search.py b/search.py
index 802129d..c44d13b 100644
--- a/search.py
+++ b/search.py
@@ -23,7 +23,7 @@ learning_rate_values = list(np.round(np.logspace(-5, -3, 21), 5))
 alpha_values = [0, 1, 2]
 widths = [64, 128, 256, 512]
 # learning_rate_values = [5e-4]
-batch_size_values = [32, 64, 128]
+batch_size_values = [8192]
 max_epochs_values = [50]
 seeds = list(range(21, 1992))