Update of unit tests