Dodałem ify

Signed-off-by: Grzes <kabza.grzegorz@gmail.com>