Fix tests
Remove unused imports
Remove unnecessary warning suppression