Fixed the issue with running integration tests in test phase.
Removed unused import.