Add missing `assert` statement