Add missing assert