diff --git a/Makefile b/Makefile
index b8a2071b..4afa1b88 100644
--- a/Makefile
+++ b/Makefile
@@ -124,6 +124,10 @@ ifeq ($(CONFIG_ACPICA),y)
 COMMON_FLAGS += -DKTF_ACPICA
 endif
 
+ifneq ($(UNITTEST),)
+COMMON_FLAGS += -DKTF_UNIT_TEST
+endif
+
 AFLAGS  := $(COMMON_FLAGS) -DASM_FILE -D__ASSEMBLY__ -nostdlib -nostdinc
 CFLAGS  := $(COMMON_FLAGS) -std=gnu99 -O2 -g -Wall -Wextra -ffreestanding -nostdlib -nostdinc
 CFLAGS  += -mno-red-zone -mno-mmx -mno-sse -mno-sse2
diff --git a/tests/unittests.c b/tests/unittests.c
index 954511f1..7fa4b014 100644
--- a/tests/unittests.c
+++ b/tests/unittests.c
@@ -173,10 +173,8 @@ int unit_tests(void *_unused) {
     schedule_task(task_user1, get_bsp_cpu());
     schedule_task(task_user2, get_cpu(1));
 
-#ifdef UNITTEST_LONGMODE
     printk("Long mode to real mode transition:\n");
     long_to_real();
-#endif
 
     return 0;
 }