clean up following pylint