Small bug fix on an assert.