[refactor] add assert statement.

add assert statement