Add braces to avoid dangling else