Update test_cse.py

Correct error from rebase.