Update cse_main.py

Correct error from rebase.