Fix missing assert

/to @raymondfeng