diff --git a/src/mem/dramsim2.cc b/src/mem/dramsim2.cc
index 6d0b84336..64acc5b57 100644
--- a/src/mem/dramsim2.cc
+++ b/src/mem/dramsim2.cc
@@ -349,7 +349,7 @@ DRAMSim2::getSlavePort(const std::string &if_name, PortID idx)
     }
 }
 
-unsigned int
+DrainState
 DRAMSim2::drain()
 {
     // check our outstanding reads and writes and if any they need to