Removed the wrong import and renaming the test