diff --git a/src/preprocessing.py b/src/preprocessing.py
index bbd4d0d..56a3543 100644
--- a/src/preprocessing.py
+++ b/src/preprocessing.py
@@ -4,11 +4,12 @@ from pandas import read_table
 def read_header(filename):
     with open(filename, "r") as f:
         header = f.readline().split()
-    return int(header[0]), int(header[1])
+    return int(header[0])
 
 
 def parse_file(filename):
-    n, m = read_header(filename)
+    n = read_header(filename)
+    m = 50
     df = read_table(
         filename, names=["source", "destination", "distance"], sep=" ", skiprows=[0]
     )