diff --git a/cast5/cast5.c b/cast5/cast5.c
index 5e318cf..00d92ae 100644
--- a/cast5/cast5.c
+++ b/cast5/cast5.c
@@ -27,15 +27,16 @@
  * 
  */
  
- #include <stdint.h>
- #include <string.h>
- #include "cast5.h"
+#include <stdint.h>
+#include <string.h>
+#include "cast5.h"
+#include <avr/pgmspace.h> 
+
+#undef DEBUG
  
- #undef DEBUG
- 
- #ifdef DEBUG
-  #include "cli.h"
- #endif
+#ifdef DEBUG
+ #include "cli.h"
+#endif
  
 #include "cast5-sbox.h"