TW-57802 Fix tests