fixes for  TEZ-1130 & TEZ-1076