diff --git a/src/rust/modrm.rs b/src/rust/modrm.rs
index 3f0956ac..2d59b23a 100644
--- a/src/rust/modrm.rs
+++ b/src/rust/modrm.rs
@@ -262,7 +262,7 @@ pub fn jit_add_seg_offset(ctx: &mut JitContext, default_segment: u32) {
     }
     codegen::gen_profiler_stat_increment(ctx.builder, profiler::stat::SEG_OFFSET_NOT_OPTIMISED);
 
-    if cfg!(debug_assertions) && seg != CS && seg != SS {
+    if cfg!(feature = "profiler") && seg != CS && seg != SS {
         ctx.builder.const_i32(seg as i32);
         ctx.builder.call_fn1("log_segment_null");
     }