Fixing compilation errors. Braces were wrong.