These two lines were removed to match example_discrete.py, which was, I believe, correct.