diff --git a/shell/cmdedit.h b/shell/cmdedit.h
index b96ffd3..0ab94c6 100644
--- a/shell/cmdedit.h
+++ b/shell/cmdedit.h
@@ -1,13 +1,14 @@
 #ifndef CMDEDIT_H
 #define CMDEDIT_H
 
-#ifdef BB_FEATURE_SH_COMMAND_EDITING
+#ifdef BB_FEATURE_COMMAND_EDITING
 #include <stddef.h>
+#include "busybox.h"
 
 void cmdedit_init(void);
 void cmdedit_terminate(void);
 void cmdedit_read_input(char* promptStr, char* command);		/* read a line of input */
 
-#endif /* BB_FEATURE_SH_COMMAND_EDITING */
+#endif /* BB_FEATURE_COMMAND_EDITING */
 
 #endif /* CMDEDIT_H */