Rewrite the test to respect the established style