Fix 'if' statement