diff --git a/risc_v/src/main.rs b/risc_v/src/main.rs
index ab359fc..3d4b285 100755
--- a/risc_v/src/main.rs
+++ b/risc_v/src/main.rs
@@ -192,5 +192,7 @@ pub mod syscall;
 pub mod trap;
 pub mod uart;
 pub mod virtio;
+#[cfg(test)]
+pub mod test;