Add missing braces so the test works properly.