Fixes some lack of braces on one liners