Fixed unused import warning in tests