For ENG-5311, fix test