Add missing test class