Fixes more deprecated imports

Forgot to update the tests project.