Updated unit tests for ts type