diff --git a/include/nwnss/include/schedule.h b/include/nwnss/include/schedule.h
index f35106a..b5844c5 100644
--- a/include/nwnss/include/schedule.h
+++ b/include/nwnss/include/schedule.h
@@ -47,12 +47,11 @@
 #	include <public/zOmni.h>
 #endif
 
-#ifdef __linux__
+#ifdef MARS_NWE_NWNSS_USERSPACE
 #ifndef _PSSMPK_H_
 #	include <include/pssmpk.h>
 #endif
 
-#ifdef MARS_NWE_NWNSS_USERSPACE
 /* nssOSAPIs.h may have been included first and defines these as Linux-kernel
  * macros.  The userspace scheduler port provides real functions instead. */
 # undef ZOS_Sleep
@@ -61,8 +60,6 @@
 # undef ZOS_ScheduleWorkToDo
 # undef ZOS_ScheduleFastWorkToDo
 # undef ZOS_CancelWorkToDo
-#endif
-
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -495,6 +492,6 @@ typedef struct zWorkProc2_s
 }
 #endif
 
-#endif /* __linux__ */
+#endif /* MARS_NWE_NWNSS_USERSPACE */
 
 #endif /* _SCHEDULE_H_ */