diff --git a/openair1/PHY/CODING/nr_compute_tbs.c b/openair1/PHY/CODING/nr_compute_tbs.c
index d995befdb5..4a1f213d73 100644
--- a/openair1/PHY/CODING/nr_compute_tbs.c
+++ b/openair1/PHY/CODING/nr_compute_tbs.c
@@ -23,7 +23,7 @@
    purpose: Compute NR TBS
    author: Hongzhi WANG (TCL)
 */
-//#include "PHY/defs.h"
+#include "PHY/defs_nr_UE.h"
 //#include "SCHED/extern.h"
 
 #define INDEX_MAX_TBS_TABLE (93)
diff --git a/openair1/PHY/CODING/nr_segmentation.c b/openair1/PHY/CODING/nr_segmentation.c
index e14d04df22..ad2e809815 100644
--- a/openair1/PHY/CODING/nr_segmentation.c
+++ b/openair1/PHY/CODING/nr_segmentation.c
@@ -24,7 +24,7 @@
    author: Hongzhi WANG (TCL)
    date: 12.09.2017
 */
-//#include "PHY/defs.h"
+#include "PHY/defs_nr_UE.h"
 //#include "SCHED/extern.h"
 
 //#define DEBUG_SEGMENTATION