1 files changed, 3 insertions, 2 deletions
diff --git a/strndup.c b/strndup.c
index cc828a5..e2f144f 100644
--- a/strndup.c
+++ b/strndup.c
@@ -1,5 +1,6 @@
 /* See LICENSE file for copyright and license details. */
 #include "libsimple.h"
+#ifndef TEST
 
 
 char *
@@ -20,8 +21,8 @@ libsimple_strndup(const char *s, size_t n) /* FIXME */
 }
 
 
-#ifdef TEST
-#include <assert.h>
+#else
+#include "test.h"
 
 int
 main(void)