1 files changed, 2 insertions, 2 deletions
diff --git a/src/mem/dramsim2.cc b/src/mem/dramsim2.cc
index be244bd4a..dfeca0d47 100644
--- a/src/mem/dramsim2.cc
+++ b/src/mem/dramsim2.cc
@@ -362,11 +362,11 @@ DRAMSim2::drain(DrainManager* dm)
     // check our outstanding reads and writes and if any they need to
     // drain
     if (nbrOutstanding() != 0) {
-        setDrainState(Drainable::Draining);
+        setDrainState(DrainState::Draining);
         drainManager = dm;
         return 1;
     } else {
-        setDrainState(Drainable::Drained);
+        setDrainState(DrainState::Drained);
         return 0;
     }
 }