fixes test issue and pep8 compliance.

because pep8