fixed broken test
fixed pylint errors