Removed import replace_keys