Add tests fixes