diff --git a/src/memory.py b/src/memory.py
index bd745d8..f20d05e 100644
--- a/src/memory.py
+++ b/src/memory.py
@@ -16,9 +16,9 @@ def sizeof_fmt(num, suffix='B'):
         num /= 1024.0
     return "%.1f %s%s" % (num, 'Yi', suffix)
 
-def print_largest_variables(top=10):
+def print_largest_variables(variables, top=10):
     """ Print the names and sizes of the top largest variables. """
-    variables = list(locals().items())
+    variables = list(variables.items())
     variables_with_sizes = ((name, sys.getsizeof(value)) for name, value in variables)
 
     if not top: