Fix for regression in commit 1fbd95f7 which would cause

salloc/sbatch -w node0 -n2 to allocate 2 nodes instead of only 1 if it
could.

This makes the code look like it does in srun which does do the correct
thing.