diff --git a/src/register/hei_register.hpp b/src/register/hei_register.hpp
index e03e819..be769f3 100755
--- a/src/register/hei_register.hpp
+++ b/src/register/hei_register.hpp
@@ -17,11 +17,11 @@
 //  Includes
 //----------------------------------------------------------------------
 
-#include <iipbits.h>
-#include <iipconst.h>
+#include <hei_includes.hpp>
+#include <util/hei_bit_string.hpp>
+
 #include <iipsdbug.h>
 #include <prdfMain.H>
-#include <prdfTrace.H>
 
 namespace libhei
 {