Fixed two flake8 errors

Fixed two flake8 errors