Fixed a bug with an experiment I was trying out