Update experiment.py

A line got dropped out at some point.