Подскажите, правило написания "по окончании" и "по окончанию"

Предлог со значением "после" - по окончании: по окончании мероприятия гости разошлись; но: судить о форме слова по окончанию.