minor fix, bring back the assertion.