Remove unneeded import
Add another test case