diff --git a/lib/trace.c b/lib/trace.c
index 831283c..defc971 100644
--- a/lib/trace.c
+++ b/lib/trace.c
@@ -57,7 +57,7 @@
 	return offset / FUNC_SITE_SIZE;
 }
 
-#if defined(CONFIG_EFI_LOADER) && defined(CONFIG_ARM)
+#if defined(CONFIG_EFI_LOADER) && (defined(CONFIG_ARM) || defined(CONFIG_RISCV))
 
 /**
  * trace_gd - the value of the gd register