Remove unused import that caused tests to fail