diff --git a/memory_profiler.py b/memory_profiler.py
index 6d26422..53cfe84 100644
--- a/memory_profiler.py
+++ b/memory_profiler.py
@@ -676,12 +676,13 @@ def trace(self, code, lineno, prev_lineno):
         previous_inc = prev_value[0] if prev_value else 0
 
         prev_line_value = self[code].get(prev_lineno, None) if prev_lineno else None
-        prev_line_memory = prev_line_value[1] if prev_line_value else 0
+        prev_line_memory = prev_line_value[3] if prev_line_value else 0
         occ_count = self[code][lineno][2] + 1 if lineno in self[code] else 1
         self[code][lineno] = (
             previous_inc + (memory - prev_line_memory),
             max(memory, previous_memory),
             occ_count,
+            memory,
         )
 
     def items(self):