If block fix