Add Changes in tests