Fixed a missing brace issue