Update unit test to support Sensitive data type