Minor bugs from flake8