Fixed error (removed unused import)