fix traceback issue
fix for pylint