Added missing braces that broke script.