fix migration count assertion failure

Introduced by my previous commit