Forgot to change an import after renaming test file