Fix end of block brace bug