Fixed test case file & removed unused imports