Small bug in experiment.py