Оба варианта являются правильными. В данном случае в русском языке можно использовать как "по окончании", так и "по окончанию". Нет строгого правила, какой из этих вариантов использовать, они равнозначны и корректны.