small fix
pep 8 cleanup