diff --git a/crawler/rec/floorplan.py b/crawler/rec/floorplan.py
index b942c76..562ab08 100644
--- a/crawler/rec/floorplan.py
+++ b/crawler/rec/floorplan.py
@@ -17,9 +17,9 @@ def inference(image_path):
     return output, predictions
 
 
-def extract_total_sqm(deplot_input_str):
+def extract_total_sqm(input_str: str):
     sqmregex = r"(\d+\.\d*) ?(sq ?m|sq. ?m)"
-    matches = re.findall(sqmregex, deplot_input_str.lower())
+    matches = re.findall(sqmregex, input_str.lower())
     if len(matches) == 0:
         return None
     sqms = [float(m[0]) for m in matches]