Remove extra import from run_experiment