Move json import to top. Fixed load_json