Remove unnecessary import from tests