Add support for assertions