Tests clean up

- Removed unnecessary tests.