Может быть эти строки засунуть внутрь if'а?