Add missing test snippet