Put back missing if statements