diff --git a/DRAMSys/traceAnalyzer/scripts/memUtil.py b/DRAMSys/traceAnalyzer/scripts/memUtil.py
index b7136faa..049e707a 100755
--- a/DRAMSys/traceAnalyzer/scripts/memUtil.py
+++ b/DRAMSys/traceAnalyzer/scripts/memUtil.py
@@ -64,16 +64,14 @@ def getWindowSize(connection):
 
 
 def maximum_data_rate(connection):
-    memspec = MemSpec(connection)
-    memoryType = memspec.getValue("memoryType")
-    if (memoryType.find("DDR") != -1):
-        width = 64
-    else:
-        if (memoryType.find("WIDEIO") != -1):
-            width = memspec.getValue("width")
+    memspec = MemSpec(connection)    
+    try:
+        width = memspec.getIntValue("memarchitecturespec", "nbrOfDevicesOnDIMM") * memspec.getIntValue("memarchitecturespec", "width")
+    except:
+        width = memspec.getIntValue("memarchitecturespec", "width")
     clk = memspec.getIntValue("memtimingspec", "clkMhz")
     rate = memspec.getIntValue("memarchitecturespec", "dataRate")
-    maxDataRate = float(clk)*float(width)*float(rate)
+    maxDataRate = float(clk) * float(width) * float(rate)
     return maxDataRate