Add assertions.

Add assertions.