diff --git a/dram/src/simulation/Dram.h b/dram/src/simulation/Dram.h
index 3bff3ccb..0e429f04 100644
--- a/dram/src/simulation/Dram.h
+++ b/dram/src/simulation/Dram.h
@@ -21,11 +21,13 @@
 #include "../common/TlmRecorder.h"
 #include "../common/third_party/DRAMPower/src/libdrampower/LibDRAMPower.h"
 #include "../common/third_party/DRAMPower/src/xmlparser/MemSpecParser.h"
-
+#include "../common/third_party/DRAMPower/src/MemorySpecification.h"
+#include "../common/third_party/DRAMPower/src/MemCommand.h"
 
 using namespace std;
 using namespace tlm;
 using namespace core;
+using namespace Data;
 
 template<unsigned int BUSWIDTH = 128, unsigned int WORDS = 4096, bool STORE = true, bool FIXED_BL = false,
         unsigned int FIXED_BL_VALUE = 0>
@@ -45,9 +47,6 @@ struct Dram: sc_module
 
     ~Dram()
     {
-
-
-
         DRAMPower->getEnergy();
         cout << "Total Energy" << "\t" << DRAMPower->mpm.energy.total_energy << endl;
         cout << "Average Power" << "\t" << DRAMPower->mpm.power.average_power << endl;