Add missing asserts