diff --git a/dpdispatcher/machines/pbs.py b/dpdispatcher/machines/pbs.py
index 23d9ff76..ac9cf366 100644
--- a/dpdispatcher/machines/pbs.py
+++ b/dpdispatcher/machines/pbs.py
@@ -221,10 +221,8 @@ def gen_script_header(self, job):
                 f"#$ -q {resources.queue_name}"
             )
 
-        if (
-            resources["strategy"].get("customized_script_header_template_file")
-            is not None
-        ):
+        if (resources["strategy"].get("customized_script_header_template_file")
+                is not None):
             filename = resources["strategy"]["customized_script_header_template_file"]
             sge_script_header = customized_script_header_template(filename, resources)
         else:
diff --git a/tests/sample_class.py b/tests/sample_class.py
index 3a0b5492..94a95bbd 100644
--- a/tests/sample_class.py
+++ b/tests/sample_class.py
@@ -24,6 +24,7 @@ def get_sample_resources(cls):
             cpu_per_node=4,
             gpu_per_node=1,
             queue_name="T4_4_15",
+            sge_pe_name= "mpi",
             group_size=2,
             custom_flags=[],
             strategy={"if_cuda_multi_devices": False},