Removed unused import from first attempt at fix