diff --git a/src/plugins/task/affinity/affinity.c b/src/plugins/task/affinity/affinity.c
index cb1d276436239540a8b25df16d8338729b4a1fb6..3c1958d4fa62b6c1b3e0b3e66a336894cdcf4c50 100644
--- a/src/plugins/task/affinity/affinity.c
+++ b/src/plugins/task/affinity/affinity.c
@@ -397,7 +397,7 @@ int slurm_setaffinity(pid_t pid, size_t size, const cpu_set_t *mask)
 #ifdef __FreeBSD__
         rval = cpuset_setaffinity(CPU_LEVEL_WHICH, CPU_WHICH_PID,
 				pid, size, mask);
-#elif SCHED_GETAFFINITY_THREE_ARGS
+#elif defined(SCHED_GETAFFINITY_THREE_ARGS)
 	rval = sched_setaffinity(pid, size, mask);
 #else
 	rval = sched_setaffinity(pid, mask);