Fix missing price test error
pep8 fixes