Update code to please Travis CI
Add curly braces to if statement.