Fixes an outdated import issue.