add changes in tests