Removed import and a commented out a test