fix the tests

Signed-off-by: Chris “Kwpolska” Warrick <kwpolska@gmail.com>
flake8

Signed-off-by: Chris “Kwpolska” Warrick <kwpolska@gmail.com>