Add missing type param for test