Fix test
Remove unnecessary package