Add asserts