Add missing assertions