fix: import fix for drive refactoring

[QuantumExplorer]

Issue being fixed or feature implemented

cargo test stopped working because of some issues in imports.

What was done?

They are now fixed.

How Has This Been Tested?

Breaking Changes

Checklist:

• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas
• I have added or updated relevant unit/integration/functional/e2e tests
• I have added "!" to the title and described breaking changes in the corresponding section if my code contains any
• I have made corresponding changes to the documentation if needed

For repository code-owners and collaborators only

• I have assigned this pull request to a milestone

[QuantumExplorer]

Merging this in as it is just a small but critical fix for tests.