Add missing test methods