Тогда лучше взять билет обратный на дату окончания ?

Да, лучше вылететь в день окончания визы. А то мало ли, вдруг у пограничника будет плохое настроение