Remove unnecessary imports from test code