Fix missing "if" statement