Add back missing brace