Update Unit test for custom type.