Add some asserts