[Update] Unit tests