Add tests about update