Fix code style

Added braces around if statement as per build failure.