code style
better assertions