diff --git a/src/pam_panic_pw/pam_panic_pw.c b/src/pam_panic_pw/pam_panic_pw.c
index 460a280..db9fe2b 100644
--- a/src/pam_panic_pw/pam_panic_pw.c
+++ b/src/pam_panic_pw/pam_panic_pw.c
@@ -21,9 +21,9 @@ LICENSE :      GNU-GPLv3
 
 #define _(String) gettext(String)
 
-int writePasswords(char pw[][99]){
+int writePasswords(char pw[][99], char* pwfile){
 
-  FILE *f = fopen(PPASSFILE, "w");
+  FILE *f = fopen(pwfile, "w");
   if(f == NULL){
     fprintf(stderr, _("ERROR opening file!\n"));
     return 2;
@@ -33,11 +33,12 @@ int writePasswords(char pw[][99]){
 
   fclose(f);
 
-  chmod(PPASSFILE, 0644);
+  chmod(pwfile, 0644);
 
   return 0;
 }
 
+#ifndef TEST
 
 int main(void){
 
@@ -105,6 +106,6 @@ int main(void){
 
   /* Save the results. */
   
-  return writePasswords(pw);
+  return writePasswords(pw, PPASSFILE);
 }
-
+#endif
diff --git a/src/pam_panic_pw/pam_panic_pw.h b/src/pam_panic_pw/pam_panic_pw.h
index d75a6b3..bfb98c0 100644
--- a/src/pam_panic_pw/pam_panic_pw.h
+++ b/src/pam_panic_pw/pam_panic_pw.h
@@ -5,4 +5,4 @@ DATE :         2018-03-27T02:34:08+02:00
 LICENSE :      GNU-GPLv3
 */
 
-int writePassword(char pw[][99]);
+int writePasswords(char pw[][99], char* pwfile);