python tool: added missing import statement