removed outdated import
fixed test failures