Fix for load_results plotting script