Add missing unittest