removed unnecessary imports from unit test