New Updates in unit tests