BUG: also remove pyplot top level import