Update test.py

I'm sure this line is supposed to be conditional on the else.