added missing import of six