Update missed unit tests