Added braces to follow consistent syntax

Added braces to follow consistent syntax in the test case.