diff --git a/litedram/frontend/bist.py b/litedram/frontend/bist.py
index 0480ad5..f0c7f77 100644
--- a/litedram/frontend/bist.py
+++ b/litedram/frontend/bist.py
@@ -30,7 +30,7 @@ class LFSR(Module):
         ]
 
 
-class LiteDRAMBISTGenerator(Module):
+class LiteDRAMBISTGenerator(Module, AutoCSR):
     def __init__(self, dram_port):
         self.reset = CSR()
         self.shoot = CSR()
@@ -66,7 +66,7 @@ class LiteDRAMBISTGenerator(Module):
         ]
 
 
-class LiteDRAMBISTChecker(Module):
+class LiteDRAMBISTChecker(Module, AutoCSR):
     def __init__(self, dram_port):
         self.reset = CSR()
         self.shoot = CSR()