diff --git a/python/pythonrc.py b/python/pythonrc.py
index 4594c20..c52b57b 100644
--- a/python/pythonrc.py
+++ b/python/pythonrc.py
@@ -17,4 +17,8 @@
 
 readline.set_history_length(1024 * 5)
 
-atexit.register(lambda x=history: readline.write_history_file(x))
+def write_history(x):
+  import readline:
+  readline.write_history_file(x)
+
+atexit.register(write_history)