diff --git a/eessi/testsuite/tests/apps/osu.py b/eessi/testsuite/tests/apps/osu.py
index 5e484076..95a82a18 100644
--- a/eessi/testsuite/tests/apps/osu.py
+++ b/eessi/testsuite/tests/apps/osu.py
@@ -102,8 +102,9 @@ def set_num_tasks_per_node(self):
     @run_after('setup')
     def set_num_cpus_per_task(self):
         """ Since num_tasks_per_node is already set. This function sets
-        num_cpus_per_task for 1 node and 2 node options. For """
-        if(SCALES.get(self.scale).get('num_nodes') == 1 and
+        num_cpus_per_task for 1 node and 2 node options where the request is
+        for full nodes."""
+        if(SCALES.get(self.scale).get('num_nodes') >= 1 and
            SCALES.get(self.scale).get('node_part', 0) == 1):
             hooks.assign_one_task_per_compute_unit(self,
                                                    COMPUTE_UNIT.get(CPU, 'cpu'))