Fix tests
Formatting