Fixed outdated import in pylabplot