Fix expectation
Pep8