diff --git a/test.c b/test.c
index 01fe109..84adb63 100644
--- a/test.c
+++ b/test.c
@@ -30,7 +30,16 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+/* These #defines must be present according to PAM documentation. */
+#define PAM_SM_AUTH
+
+#ifdef HAVE_SECURITY_PAM_APPL_H
 #include <security/pam_appl.h>
+#endif
+#ifdef HAVE_SECURITY_PAM_MODULES_H
+#include <security/pam_modules.h>
+#endif
+
 
 int
 main (int argc, char **argv)