Add missing assert.
Merge pull request #961 from ktf/add-missing-asserts

Misc fixes -- Add missing assert.