1 files changed, 2 insertions, 1 deletions
diff --git a/dev/io_device.hh b/dev/io_device.hh
index 39e6fa4aa..9300d87e7 100644
--- a/dev/io_device.hh
+++ b/dev/io_device.hh
@@ -34,6 +34,7 @@
 class BaseInterface;
 class Bus;
 class HierParams;
+template <class Bus> class DMAInterface;
 
 class PioDevice : public FunctionalMemory
 {
@@ -48,7 +49,7 @@ class PioDevice : public FunctionalMemory
 class DmaDevice : public PioDevice
 {
   protected:
-    BaseInterface *dmaInterface;
+    DMAInterface<Bus> *dmaInterface;
 
   public:
     DmaDevice(const std::string &name);