diff --git a/mimic_direct_extract.py b/mimic_direct_extract.py
index c65e683..577ea41 100644
--- a/mimic_direct_extract.py
+++ b/mimic_direct_extract.py
@@ -231,7 +231,7 @@ def save_numerics(
 
     var_map = var_map[
         ['LEVEL2', 'ITEMID', 'LEVEL1']
-    ].rename_axis(
+    ].rename(
         {'LEVEL2': 'LEVEL2', 'LEVEL1': 'LEVEL1', 'ITEMID': 'itemid'}, axis=1
     ).set_index('itemid')