diff --git a/main.py b/main.py
index 80a669339562ebcf147cd7613a6e4715cd158e1e..778db82b7addf5e71dd2fbaafffc7697fa72885f 100644
--- a/main.py
+++ b/main.py
@@ -43,7 +43,8 @@ def main(set_identifier_bit_count, line_count, block_identifier_bit_count, file_
     b = block_identifier_bit_count
     B = block_size = int(math.pow(2, b))
     m = address_identifier_bit_count = E * S
-    M = total_bytes = int(math.pow(2, m))
+    # M = total_bytes = int(math.pow(2, m))
+    M = int(math.pow(2, 64))
     C = capacity = S * E * B
     tag_identifier_bit_count = m - (b + s)
 
@@ -107,7 +108,7 @@ def handle_line(cache, line, M, B, S, hits, misses, evictions):
     # Check if instruction load. If so, we can skip line.
     if split_line[0] == 'I':
         # Found instruction load. Returning.
-        return (hits, misses, evictions)
+        return (cache, hits, misses, evictions)
     else:
         line = line.strip()
         conflict_miss = -1