diff --git a/benchmark/tgt_ijcai_benchmark.py b/benchmark/tgt_ijcai_benchmark.py
index 33f23e2..29abc2e 100644
--- a/benchmark/tgt_ijcai_benchmark.py
+++ b/benchmark/tgt_ijcai_benchmark.py
@@ -231,7 +231,7 @@ def create_edge(data, behavior, beh_idx, pkey_name, pkey_idx):
 elif args.model in ["contextgnn", "shallowrhsgnn"]:
     model_search_space = {
         "encoder_channels": [32, 64, 128, 256],
-        "encoder_layers": [2, 4, 8],
+        "encoder_layers": [2, 4],
         "channels": [32, 64, 128, 256],
         "embedding_dim": [32, 64, 128, 256],
         "norm": ["layer_norm", "batch_norm"],