Restored missing "if" statement.