Fix missing assert module

/to @raymondfeng