Update test_control_structures.py

should work ?