Updates in unittest