Fixed outdated import