added missing import
small rename