Fix tests to match styles