Как мне переустановить lax.js после того, как smoothstate вставит новый контент?

Question

Я использовал последнюю версию smoothstate и lax.js - reinit lax в smoothstate onAfter приводил к ошибкам в анимации lax css. Как правильно переустановить lax.js после того, как smoothstate вставит новый контент?

Я пытался lax.setup() в onAfter функция плавного состояния

onAfter: function($container, $newContent) {
    $('html').removeClass('animate');
    lax.setup();
}