diff --git a/src/main/java/li/cil/oc2/client/gui/RISCVTestScreen.java b/src/main/java/li/cil/oc2/client/gui/RISCVTestScreen.java
index 13dc8fe0..f7ca21f8 100644
--- a/src/main/java/li/cil/oc2/client/gui/RISCVTestScreen.java
+++ b/src/main/java/li/cil/oc2/client/gui/RISCVTestScreen.java
@@ -199,6 +199,7 @@ public final class RISCVTestScreen extends Screen {
 
         loadProgramFile(memory, Buildroot.getLinuxImage());
         loadProgramFile(rom, Buildroot.getFirmware());
+        board.installDeviceTree(0x80000000 + 0x02200000);
 
         runner = new ConsoleRunner(board);
     }
diff --git a/src/main/java/li/cil/oc2/common/tile/ComputerTileEntity.java b/src/main/java/li/cil/oc2/common/tile/ComputerTileEntity.java
index a5a3bad3..80c73cf3 100644
--- a/src/main/java/li/cil/oc2/common/tile/ComputerTileEntity.java
+++ b/src/main/java/li/cil/oc2/common/tile/ComputerTileEntity.java
@@ -208,6 +208,7 @@ public final class ComputerTileEntity extends TileEntity implements ITickableTil
         }
 
         virtualMachine.board.reset();
+        virtualMachine.board.installDeviceTree(0x80000000 + 0x02200000);
         runner = new ConsoleRunner(virtualMachine);
     }