Setup tests
Fix tests.