Fix end-brace behavior