fix tests
Remove unnecessary functionality